Add test for order as tuple

radionets-project · Oct 25, 2024 · 4edfd60 · 4edfd60
1 parent d4522f9
commit 4edfd60
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/test_simulation.py b/tests/test_simulation.py
@@ -218,6 +218,14 @@ def test_polarisation_field_order(self):
         # assert order = 1 and order = [1, 1] yield same images
         assert_array_equal(pf, pf_ref, strict=True)
 
+        pf = self.pol.rand_polarisation_field(
+            shape=self.im_shape,
+            random_state=42,
+            order=(1, 1),
+        )
+        # assert order = (1, 1) and order = [1, 1] yield same images
+        assert_array_equal(pf, pf_ref, strict=True)
+
         pf = self.pol.rand_polarisation_field(
             shape=self.im_shape,
             random_state=42,