1 files changed, 71 insertions, 0 deletions
diff --git a/tests/test_classic_visualizer.py b/tests/test_classic_visualizer.py
new file mode 100644
index 0000000..e301263
--- /dev/null
+++ b/tests/test_classic_visualizer.py
@@ -0,0 +1,71 @@
+from avp.command import Command
+from pytestqt import qtbot
+from pytest import fixture
+from . import audioData, MockSignal, imageDataSum
+
+
+sampleSize = 1470  # 44100 / 30 = 1470
+
+
+@fixture
+def coreWithClassicComp(qtbot):
+    """Fixture providing a Command object with Classic Visualizer component added"""
+    command = Command()
+    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: coreWithClassicComp.selectedComponents[0].transformData(
+            0, audioData[0], sampleSize, 0.08, 0.8, None, 20
+        )
+    }
+    for i in range(sampleSize, len(audioData[0]), sampleSize):
+        spectrumArray[i] = coreWithClassicComp.selectedComponents[0].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