added suport for chunk read of clusters

2023-06-05 10:20:14 +02:00
parent 8aa1b6bcbd
commit 221f7e0c0d
4 changed files with 61 additions and 12 deletions
--- a/tests/test_ClusterReader.py
+++ b/tests/test_ClusterReader.py
@ -114,4 +114,11 @@ def test_read_file_with_noise_mask(data_path):
    noise_cut[133,80] = 0
    r = ClusterFileReader(fname)
    cl = r.read(85, noise_cut)
-    assert cl.size == 10
+    assert cl.size == 10
+
+def test_chunk_config(data_path):
+    fname= data_path/'noise_test.clust'
+    #File contains total 70 clusters
+    r = ClusterFileReader(fname, chunk = 5)
+    assert r.read().size == 5
+    assert r.read(10).size == 10