1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
from avp.command import Command
from avp.toolkit.visualizer import transformData
from pytestqt import qtbot
from pytest import fixture
from . import audioData, command, MockSignal, imageDataSum
sampleSize = 1470 # 44100 / 30 = 1470
@fixture
def coreWithClassicComp(qtbot, command):
"""Fixture providing a Command object with Classic Visualizer component added"""
command.core.insertComponent(
0, command.core.moduleIndexFor("Classic Visualizer"), command
)
yield command.core
def test_comp_classic_added(coreWithClassicComp):
"""Add Classic Visualizer to core"""
assert len(coreWithClassicComp.selectedComponents) == 1
def test_comp_classic_removed(coreWithClassicComp):
"""Remove Classic Visualizer from core"""
coreWithClassicComp.removeComponent(0)
assert len(coreWithClassicComp.selectedComponents) == 0
def test_comp_classic_drawBars(coreWithClassicComp, audioData):
"""Call drawBars after creating audio spectrum data manually."""
spectrumArray = {0: transformData(0, audioData[0], sampleSize, 0.08, 0.8, None, 20)}
for i in range(sampleSize, len(audioData[0]), sampleSize):
spectrumArray[i] = transformData(
i,
audioData[0],
sampleSize,
0.08,
0.8,
spectrumArray[i - sampleSize].copy(),
20,
)
image = coreWithClassicComp.selectedComponents[0].drawBars(
1920, 1080, spectrumArray[sampleSize * 4], (0, 0, 0), 0
)
assert imageDataSum(image) == 37872316
def test_comp_classic_drawBars_using_preFrameRender(coreWithClassicComp, audioData):
"""Call drawBars after creating audio spectrum data using preFrameRender."""
comp = coreWithClassicComp.selectedComponents[0]
comp.preFrameRender(
completeAudioArray=audioData[0],
sampleSize=sampleSize,
progressBarSetText=MockSignal(),
progressBarUpdate=MockSignal(),
)
image = comp.drawBars(
1920,
1080,
coreWithClassicComp.selectedComponents[0].spectrumArray[sampleSize * 4],
(0, 0, 0),
0,
)
assert imageDataSum(image) == 37872316
|
