Check that the incorrect value raises an error

cmelab · Oct 26, 2021 · 73ba2b6 · 73ba2b6
1 parent e352da2
commit 73ba2b6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gixstapose/tests/test_scene.py b/gixstapose/tests/test_scene.py
@@ -1,6 +1,7 @@
 import os
 
 import fresnel
+import pytest
 
 from gixstapose.draw_scene import get_scene
 
@@ -36,5 +37,5 @@ def test_scene_colors():
     scene, info = get_scene(methanegsd, color="jet")
     assert type(scene) is type(fresnel.Scene())
 
-    scene, info = get_scene(methanegsd, color="heck")
-    assert type(scene) is type(fresnel.Scene())
+    with pytest.raises(ValueError):
+        get_scene(methanegsd, color="heck")