Make a test that will fail with the current grouper

batdata/extractors/tests/test_maccor.py

@@ -1,13 +1,32 @@
 """Tests related to the MACCOR parser"""
 
 import os
+
+from pytest import fixture
+
 from batdata.extractors.maccor import MACCORExtractor
 
 test_file = os.path.join(os.path.dirname(__file__), 'files', 'maccor_example.001')
 
 
-def test_validation():
+@fixture()
+def extractor():
+    return MACCORExtractor()
+
+
+def test_validation(extractor):
     """Make sure the parser generates valid outputs"""
-    extractor = MACCORExtractor()
     data = extractor.parse_to_dataframe([test_file])
     data.validate_columns(allow_extra_columns=False)
+
+
+def test_grouping(extractor, tmp_path):
+    # Make a file structure with two sets of experiments and a nonsense file
+    for f in ['README', 'testA.002', 'testA.001', 'testB.001']:
+        (tmp_path / f).write_text('junk')
+
+    # Test the grouping
+    groups = list(extractor.identify_files(tmp_path))
+    assert len(groups) == 2
+    assert (str(tmp_path / 'testA.001'), str(tmp_path / 'testA.002')) in groups
+    assert (str(tmp_path / 'testB.001'),) in groups