use np.loadtxt

c5cf9ffc · BO ZHANG · 08f044d4 · c5cf9ffc
Commit c5cf9ffc authored 1 year ago by BO ZHANG
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 2 deletions
+2 -2
--- a/tests/test_flip_image.py
+++ b/tests/test_flip_image.py
@@ -20,13 +20,13 @@ class FlipImageTestCase(unittest.TestCase):

    def test_flip_image_on_server(self):
        """test flip image on server"""
-        image_input = np.load(
+        image_input = np.loadtxt(
            os.path.join(
                os.environ["UNIT_TEST_DATA_ROOT"],
                "csst_proto/test_flip/input/image.txt",
            )
        )
-        image_answer = np.load(
+        image_answer = np.loadtxt(
            os.path.join(
                os.environ["UNIT_TEST_DATA_ROOT"],
                "csst_proto/test_flip/answer/flipped_image.txt",