critical bugfix in fiber assertion

Author: grg2rsr

iblrig/neurophotometrics.py

@@ -254,7 +254,7 @@ def _validate_neurophotometrics_description(
     # assert each location being sampled at max twice and if so, by different bands
     for location in set(locations):
         ix = [i for i, loc in enumerate(locations) if loc == location]
-        assert len(ix) == 2, 'there are only 2 possible bands'
+        assert len(ix) <= 2, 'there are only 2 possible bands'
         rois_per_loc = [rois[i] for i in ix]
         # check that each band is present only once
         assert sum(True for roi in rois_per_loc if roi.startswith('G')) == 1, 'duplicate green band'

iblrig/test/test_neurophotometrics.py

@@ -8,15 +8,15 @@ class TestExperimentDescription(unittest.TestCase):
     def test_neurophotometrics_description(self):
         dt = datetime.datetime.fromisoformat('2024-10-11T11:11:00')
         d = neurophotometrics_description(
-            rois=['Region1G', 'Region2G'], locations=['SI', 'VTA'], sync_channel=1, start_time=dt, sync_mode='bpod'
+            rois=['G0', 'G1'], locations=['SI', 'VTA'], sync_channel=1, start_time=dt, sync_mode='bpod'
         )
         dexpected = {
             'devices': {
                 'neurophotometrics': {
                     'sync_channel': 1,
                     'datetime': '2024-10-11T11:11:00',
                     'collection': 'raw_photometry_data',
-                    'fibers': {'G1': {'location': 'SI'}, 'G2': {'location': 'VTA'}},
+                    'fibers': {'G0': {'location': 'SI'}, 'G1': {'location': 'VTA'}},
                     'sync_mode': 'bpod',
                 }
             }