extend dtype test

lueckem · Sep 2, 2024 · 0c50a09 · 0c50a09
1 parent 0611e24
commit 0c50a09
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tests/tests_cnvm/test_model.py b/tests/tests_cnvm/test_model.py
@@ -103,15 +103,26 @@ def test_output_dtype(self):
         correct_dtype_list = [np.uint8, np.uint16]
 
         for num_opinions, correct_dtype in zip(num_opinions_list, correct_dtype_list):
+            # complete network
             params = CNVMParameters(
                 num_opinions=num_opinions,
                 num_agents=self.num_agents,
                 r=1,
                 r_tilde=1,
             )
-
             model = CNVM(params)
             t_max = 5
             t, x = model.simulate(t_max)
+            self.assertEqual(correct_dtype, x.dtype)
 
+            # network
+            params = CNVMParameters(
+                num_opinions=num_opinions,
+                network=nx.barabasi_albert_graph(self.num_agents, 2),
+                r=1,
+                r_tilde=1,
+            )
+            model = CNVM(params)
+            t_max = 5
+            t, x = model.simulate(t_max)
             self.assertEqual(correct_dtype, x.dtype)