(WIP): Fix doorbell tests.

synsense · Nov 8, 2024 · a2bbb4c · a2bbb4c
1 parent 189cfb8
commit a2bbb4c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_dynapcnn/test_doorbell.py b/tests/test_dynapcnn/test_doorbell.py
@@ -3,6 +3,7 @@
 It will include testing of the network equivalence, and of the correct output configuration.
 """
 
+import pytest
 import samna
 import torch
 from nirtorch.utils import sanitize_name
@@ -76,7 +77,9 @@ def test_same_result():
 def test_auto_config():
     # - Should give an error with the normal layer ordering
     dynapcnn_net = DynapcnnNetwork(snn, input_shape=input_shape, discretize=True)
-    dynapcnn_net.make_config(chip_layers_ordering=[0, 1, 2, 3, 4])
+    with pytest.raises(ValueError):
+        dynapcnn_net.make_config(chip_layers_ordering=[0, 1, 2, 3, 4])
+    dynapcnn_net.make_config(layer2core_map="auto")
 
 
 def test_was_copied():