A couple of small test additions.

bluescarni · Jun 13, 2024 · a56eafe · a56eafe
1 parent 768ebbd
commit a56eafe
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/heyoka/_test_var_integrator.py b/heyoka/_test_var_integrator.py
@@ -47,6 +47,10 @@ def test_scalar(self):
         self.assertEqual(ts.shape, (2,))
         self.assertTrue(np.all(ts == [0.0, 0.0]))
 
+        # Test that tstate cannot be written to.
+        with self.assertRaises(ValueError) as cm:
+            ta.tstate[0] = 0.5
+
         self.assertEqual(ta.get_vslice(order=0), slice(0, 2, None))
         self.assertEqual(ta.get_vslice(order=0, component=1), slice(1, 2, None))
 
@@ -168,6 +172,10 @@ def test_batch(self):
         self.assertEqual(ts.shape, (2, 2))
         self.assertTrue(np.all(ts == [[0.0, 0.0], [0.0, 0.0]]))
 
+        # Test that tstate cannot be written to.
+        with self.assertRaises(ValueError) as cm:
+            ta.tstate[0] = 0.5
+
         self.assertEqual(ta.get_vslice(order=0), slice(0, 2, None))
         self.assertEqual(ta.get_vslice(order=0, component=1), slice(1, 2, None))