From 89919a4f9525fe6b3ff993480bea4fbfc8e6fdbf Mon Sep 17 00:00:00 2001
From: BO ZHANG <bozhang@nao.cas.cn>
Date: Tue, 23 Aug 2022 15:12:49 +0800
Subject: [PATCH] added unit test

---
 README.md                | 2 +-
 csst_proto/flip_image.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 1dfb87a..79fcb6d 100644
--- a/README.md
+++ b/README.md
@@ -30,7 +30,7 @@ A simple example on how to use this package:
 
 ```python
 import numpy as np
-from csst_proto import flip_image, print_data
+from csst_proto.top_level_interface import flip_image, read_test_image
 from astropy.io import fits
 
 # read an L0 image
diff --git a/csst_proto/flip_image.py b/csst_proto/flip_image.py
index 7f7aa41..b8cb643 100644
--- a/csst_proto/flip_image.py
+++ b/csst_proto/flip_image.py
@@ -30,4 +30,4 @@ def flip_image(img: np.ndarray):
 def read_test_image():
     """ read test image """
     fp_img = PACKAGE_PATH + "/data/test_image.txt"
-    return np.loadtxt(fp_img)
+    return np.loadtxt(fp_img, dtype=int)
-- 
GitLab