@@ -71,7 +71,7 @@ def test_produce_list_indices_point_in_polygon_this_frame():
7171 points = np .array ([[0.5 , 0.5 ], [1.5 , 1.5 ], [2.5 , 2.5 ], [3.5 , 3.5 ]])
7272
7373 # Call the function under test.
74- result = streamlines .produce_list_indices_point_in_polygon_this_frame (
74+ result = streamlines ._produce_list_indices_point_in_polygon_this_frame (
7575 vertex_list , points
7676 )
7777
@@ -90,7 +90,7 @@ def test_produce_list_centroids_empty():
9090 list_indices = [(np .array ([]),)]
9191 # Dummy particle coordinate array (won't be used since indices is empty)
9292 pts = np .array ([[0 , 0 ], [1 , 1 ]])
93- result = streamlines .produce_list_centroids_this_frame (list_indices , pts )
93+ result = streamlines ._produce_list_centroids_this_frame (list_indices , pts )
9494 assert result == [None ]
9595
9696
@@ -100,7 +100,7 @@ def test_produce_list_centroids_single_square():
100100 # Choose indices that pick points [1] and [3]
101101 indices_tuple = (np .array ([1 , 3 ]),)
102102 list_indices = [indices_tuple ]
103- result = streamlines .produce_list_centroids_this_frame (list_indices , pts )
103+ result = streamlines ._produce_list_centroids_this_frame (list_indices , pts )
104104 expected = np .average (pts [[1 , 3 ]], axis = 0 )
105105 np .testing .assert_array_almost_equal (result [0 ], expected )
106106
@@ -112,7 +112,7 @@ def test_produce_list_centroids_multiple_squares():
112112 # Second square will use pts[1], pts[3] and pts[4]
113113 indices2 = (np .array ([1 , 3 , 4 ]),)
114114 list_indices = [indices1 , indices2 ]
115- result = streamlines .produce_list_centroids_this_frame (list_indices , pts )
115+ result = streamlines ._produce_list_centroids_this_frame (list_indices , pts )
116116 expected1 = np .average (pts [[0 , 2 ]], axis = 0 )
117117 expected2 = np .average (pts [[1 , 3 , 4 ]], axis = 0 )
118118 np .testing .assert_array_almost_equal (result [0 ], expected1 )
0 commit comments