From 4e107bd0007ffa825b8571e495d6ab1da8a236bb Mon Sep 17 00:00:00 2001
From: xumi1993 <gomijianxu@gmail.com>
Date: Thu, 14 Mar 2024 21:26:25 +0800
Subject: [PATCH] update src_rec

---
 .gitignore           | 4 +++-
 test/test_src_rec.py | 9 ++++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index 144b12a..46ca5b9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -130,4 +130,6 @@ dmypy.json
 .vscode
 .DS_store
 crust_model/
-*.xmf
\ No newline at end of file
+*.xmf
+
+test_src_rec_eg
\ No newline at end of file
diff --git a/test/test_src_rec.py b/test/test_src_rec.py
index 98fa403..7e68395 100644
--- a/test/test_src_rec.py
+++ b/test/test_src_rec.py
@@ -26,7 +26,14 @@ def test_subcase_04(self):
         sr.add_noise()
         sr.add_noise(shape='uniform')
 
+    def test_subcase_05(self):
+        sr = SrcRec.read(self.fname)
+        sr.select_depth([0, 10])
+        assert sr.src_points.shape[0] == 1413 and sr.rec_points.shape[0] == 11226
+
+
 if __name__ == '__main__':
     tsr = TestSrcRec()
-    tsr.test_subcase_03()
+    tsr.test_subcase_05()
+