Update test_visualize.py

jan-janssen · web-flow · commit 5fd21e96251f · 2025-03-18T12:55:16.000+01:00
diff --git a/tests/test_visualize.py b/tests/test_visualize.py
@@ -31,7 +31,7 @@ def test_get_frame(self):
         dx, counts = np.unique(
             np.diff(frame, axis=-2).squeeze().astype(int), axis=0, return_counts=True
         )
-        self.assertEqual(dx.ptp(), 1, msg="Frames not drawn along the nearest edges")
+        self.assertEqual(np.ptp(dx), 1, msg="Frames not drawn along the nearest edges")
         msg = (
             "There must be four lines along each direction"
             + " (4 x [1, 0, 0], 4 x [0, 1, 0] and 4 x [0, 0, 1])"

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ def test_get_frame(self):`
`31`	`31`	`dx, counts = np.unique(`
`32`	`32`	`np.diff(frame, axis=-2).squeeze().astype(int), axis=0, return_counts=True`
`33`	`33`	`)`
`34`		`- self.assertEqual(dx.ptp(), 1, msg="Frames not drawn along the nearest edges")`
	`34`	`+ self.assertEqual(np.ptp(dx), 1, msg="Frames not drawn along the nearest edges")`
`35`	`35`	`msg = (`
`36`	`36`	`"There must be four lines along each direction"`
`37`	`37`	`+ " (4 x [1, 0, 0], 4 x [0, 1, 0] and 4 x [0, 0, 1])"`