From f7ece26796f129153eb3ee4682accb9a359dbe06 Mon Sep 17 00:00:00 2001
From: BO ZHANG <bozhang@nao.cas.cn>
Date: Tue, 23 Aug 2022 15:02:50 +0800
Subject: [PATCH] updated read_test_image

---
 csst_proto/flip_image.py          | 12 +++++-------
 csst_proto/top_level_interface.py |  4 ++--
 2 files changed, 7 insertions(+), 9 deletions(-)

diff --git a/csst_proto/flip_image.py b/csst_proto/flip_image.py
index 109e7a1..7f7aa41 100644
--- a/csst_proto/flip_image.py
+++ b/csst_proto/flip_image.py
@@ -26,10 +26,8 @@ def flip_image(img: np.ndarray):
     return img[::-1, ::-1]
 
 
-# a demo on how to package data
-def print_data():
-    """ print out data """
-    fp_data = PACKAGE_PATH + "/data/data.csv"
-    data = table.Table.read(fp_data)
-    print(data)
-    return
+# a demo on how to read test image
+def read_test_image():
+    """ read test image """
+    fp_img = PACKAGE_PATH + "/data/test_image.txt"
+    return np.loadtxt(fp_img)
diff --git a/csst_proto/top_level_interface.py b/csst_proto/top_level_interface.py
index aa96701..d5b7afb 100644
--- a/csst_proto/top_level_interface.py
+++ b/csst_proto/top_level_interface.py
@@ -1,4 +1,4 @@
-from .flip_image import flip_image, print_data
+from .flip_image import flip_image, read_test_image
 
 
-__all__ = ["flip_image", "print_data"]
+__all__ = ["flip_image", "read_test_image"]
-- 
GitLab