1 files changed, 15 insertions, 2 deletions
diff --git a/tests/test_comp_spectrum.py b/tests/test_comp_spectrum.py
index 870185c..5dd4e2d 100644
--- a/tests/test_comp_spectrum.py
+++ b/tests/test_comp_spectrum.py
@@ -1,7 +1,12 @@
 from avp.command import Command
 from pytestqt import qtbot
 from pytest import fixture
-from . import imageDataSum, command
+from . import (
+    imageDataSum,
+    command,
+    preFrameRender,
+    audioData,
+)
 
 
 @fixture
@@ -13,7 +18,15 @@ def coreWithSpectrumComp(qtbot, command):
     yield command.core
 
 
-def test_comp_waveform_previewRender(coreWithSpectrumComp):
+def test_comp_spectrum_previewRender(coreWithSpectrumComp):
     comp = coreWithSpectrumComp.selectedComponents[0]
     image = comp.previewRender()
     assert imageDataSum(image) == 71992628
+
+
+def test_comp_spectrum_renderFrame(coreWithSpectrumComp, audioData):
+    comp = coreWithSpectrumComp.selectedComponents[0]
+    preFrameRender(audioData, comp)
+    image = comp.frameRender(0)
+    comp.postFrameRender()
+    assert imageDataSum(image) == 117