Add unit tests for bug fixes

JesseMckinzie · JesseMckinzie · commit e09b67e515b5 · 2025-01-10T11:18:43.000-07:00
diff --git a/tests/test_filepattern.py b/tests/test_filepattern.py
@@ -291,6 +291,8 @@ class TestFilePattern():
     test_generate_filepattern_data.generate_channel_data()
     test_generate_filepattern_data.generate_sorted_data()
     test_generate_filepattern_data.generate_bracket_data()
+    test_generate_filepattern_data.generate_channel_data_sc()
+    test_generate_filepattern_data.generate_channel_data_spaces()
 
     def test_file_pattern(self):
 
@@ -579,7 +581,54 @@ def test_recursive_directory_regex_fp(self):
             assert fp_data.test_recursive_directory_fp[i][0]["directory"] == result[i][0]["directory"]
             assert str(os.path.basename(fp_data.test_recursive_directory_fp[i][1][0])) == os.path.basename(result[i][1][0])
             
+    def test_recursive_directory_regex_special_character_fp(self):
+        
+        path = self.root_directory + '/test_data/recursive_data_sc'
+        
+        filepattern = '/(?P<directory>.*)/img_r{r:ddd}_c{c:ddd}.tif'
+        
+        files = fp.FilePattern(path, filepattern, recursive=True)
+
+        result = []
+
+        for file in files():
+            result.append(file)
 
+        # test that same number of files are returned
+        assert len(result) == len(fp_data.test_recursive_directory_fp) 
+
+        # test that each variable and path are equal for each file in list
+        for i in range(len(fp_data.test_recursive_directory_fp)): 
+            print(result[i])
+            assert fp_data.test_recursive_directory_fp[i][0]["r"] == result[i][0]["r"]
+            assert fp_data.test_recursive_directory_fp[i][0]["c"] == result[i][0]["c"]
+            assert fp_data.test_recursive_directory_fp[i][0]["directory"] + '_TEST' == result[i][0]["directory"]
+            assert str(os.path.basename(fp_data.test_recursive_directory_fp[i][1][0])) == os.path.basename(result[i][1][0])
+
+    def test_recursive_directory_spaces_fp(self):
+        
+        path = self.root_directory + '/test_data/recursive_data_spaces/'
+        
+        filepattern = 'img_r{r:ddd}_c{c:ddd}.tif'
+        
+        files = fp.FilePattern(path, filepattern, recursive=True)
+
+        result = []
+
+        for file in files():
+            result.append(file)
+
+        print(result)
+
+        # test that same number of files are returned
+        assert len(result) == len(fp_data.test_recursive_space) 
+
+        # test that each variable and path are equal for each file in list
+        for i in range(len(fp_data.test_recursive_space)): 
+            print(result[i])
+            assert fp_data.test_recursive_space[i][0]["r"] == result[i][0]["r"]
+            assert fp_data.test_recursive_space[i][0]["c"] == result[i][0]["c"]
+            assert str(os.path.basename(fp_data.test_recursive_space[i][1][0])) == os.path.basename(result[i][1][0])
 
     def test_recursive_multi_directory_regex_fp(self):
         
diff --git a/tests/test_filepattern_data.py b/tests/test_filepattern_data.py
@@ -647,3 +647,42 @@
 ({'c': 1, 'directory': 'TXREAD', 'r': 2}, ['/Users/jmckinzie/Documents/GitHub/filepattern-1/tests/test_data/recursive_data/TXREAD/img_r002_c001.tif']),
 ({'c': 2, 'directory': 'TXREAD', 'r': 2}, ['/Users/jmckinzie/Documents/GitHub/filepattern-1/tests/test_data/recursive_data/TXREAD/img_r002_c002.tif']),
 ]
+
+test_recursive_space = [
+    (
+        {'c': 0, 'r': 0},
+        ['img_r000_c000.tif', 'img_r000_c000.tif', 'img_r000_c000.tif']
+    ),
+    (
+        {'c': 1, 'r': 0},
+        ['img_r000_c001.tif', 'img_r000_c001.tif', 'img_r000_c001.tif']
+    ),
+    (
+        {'c': 2, 'r': 0},
+        ['img_r000_c002.tif', 'img_r000_c002.tif', 'img_r000_c002.tif']
+    ),
+    (
+        {'c': 0, 'r': 1},
+        ['img_r001_c000.tif', 'img_r001_c000.tif', 'img_r001_c000.tif']
+    ),
+    (
+        {'c': 1, 'r': 1},
+        ['img_r001_c001.tif', 'img_r001_c001.tif', 'img_r001_c001.tif']
+    ),
+    (
+        {'c': 2, 'r': 1},
+        ['img_r001_c002.tif', 'img_r001_c002.tif', 'img_r001_c002.tif']
+    ),
+    (
+        {'c': 0, 'r': 2},
+        ['img_r002_c000.tif', 'img_r002_c000.tif', 'img_r002_c000.tif']
+    ),
+    (
+        {'c': 1, 'r': 2},
+        ['img_r002_c001.tif', 'img_r002_c001.tif', 'img_r002_c001.tif']
+    ),
+    (
+        {'c': 2, 'r': 2},
+        ['img_r002_c002.tif', 'img_r002_c002.tif', 'img_r002_c002.tif']
+    )
+]
diff --git a/tests/test_generate_filepattern_data.py b/tests/test_generate_filepattern_data.py
@@ -100,6 +100,112 @@ def generate_channel_data():
 
     print("Files generated.")
 
+def generate_channel_data_sc():
+
+    MAX = 3
+
+    directory = 'test_data'
+    root_directory = os.path.dirname(os.path.realpath(__file__))
+    path = os.path.join(root_directory, directory)
+    data_path = path + '/data'
+    recursive_path = path + '/recursive_data_sc'
+
+    try:
+        os.mkdir(path)
+        os.mkdir(data_path)
+        os.mkdir(recursive_path)
+        print('Data directory created at ' + path)
+    except FileExistsError:
+        print("Data directory already exists")
+
+    try:
+        os.mkdir(data_path)
+        print('Data directory created at ' + path)
+    except FileExistsError:
+        print("Data directory already exists")
+
+    try:
+        os.mkdir(recursive_path)
+        print('Data directory created at ' + path)
+    except FileExistsError:
+        print("Data directory already exists")
+
+    channels = ['DAPI_TEST', 'TXREAD_TEST', 'GFP_TEST']
+
+    for channel in channels:
+        current_path = recursive_path + '/' + channel
+        try:
+            os.mkdir(current_path)
+        except FileExistsError:
+            print("Channel directory already exists")
+
+        for i in range(0, MAX):
+
+            str_i = str(i).zfill(3)
+
+            for j in range(0, MAX):
+
+                str_j = str(j).zfill(3)
+                
+                recursive_name = 'img_r{}_c{}.tif'.format(str_i, str_j)
+                f = open(current_path + '/' + recursive_name, 'w+')
+                f.close()
+
+    print("Files generated.")
+
+def generate_channel_data_spaces():
+
+    MAX = 3
+
+    directory = 'test_data'
+    root_directory = os.path.dirname(os.path.realpath(__file__))
+    path = os.path.join(root_directory, directory)
+    data_path = path + '/data'
+    recursive_path = path + '/recursive_data_spaces'
+
+    try:
+        os.mkdir(path)
+        os.mkdir(data_path)
+        os.mkdir(recursive_path)
+        print('Data directory created at ' + path)
+    except FileExistsError:
+        print("Data directory already exists")
+
+    try:
+        os.mkdir(data_path)
+        print('Data directory created at ' + path)
+    except FileExistsError:
+        print("Data directory already exists")
+
+    try:
+        os.mkdir(recursive_path)
+        print('Data directory created at ' + path)
+    except FileExistsError:
+        print("Data directory already exists")
+
+    channels = ['DAPI TEST', 'TXREAD TEST', 'GFP TEST']
+
+    for channel in channels:
+        current_path = recursive_path + '/' + channel
+        try:
+            os.mkdir(current_path)
+        except FileExistsError:
+            print("Channel directory already exists")
+
+        for i in range(0, MAX):
+
+            str_i = str(i).zfill(3)
+
+            for j in range(0, MAX):
+
+                str_j = str(j).zfill(3)
+
+                recursive_name = 'img_r{}_c{}.tif'.format(str_i, str_j)
+                f = open(current_path + '/' + recursive_name, 'w+')
+                f.close()
+
+    print("Files generated.")
+
 def generate_sorted_data():
     MAX = 30
     length = 0
@@ -171,6 +277,8 @@ def generate_bracket_data():
     generate_sorted_data()
     generate_text_data()
     generate_bracket_data()
+    generate_channel_data_sc
+    generate_channel_data_spaces
 
 
 MAX = 3
