update src_rec

TomoATT · Mar 14, 2024 · 4e107bd · 4e107bd
1 parent e10e049
commit 4e107bd
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -130,4 +130,6 @@ dmypy.json
 .vscode
 .DS_store
 crust_model/
-*.xmf
+*.xmf
+
+test_src_rec_eg
diff --git 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()
+