add test for single string tags

NeurodataWithoutBorders · Jul 18, 2024 · f7a56c4 · f7a56c4
1 parent 074f6b9
commit f7a56c4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/unit/test_file.py b/tests/unit/test_file.py
@@ -142,6 +142,15 @@ def test_epoch_tags(self):
         tags = self.nwbfile.epoch_tags
         self.assertEqual(set(expected_tags), set(tags))
 
+    def test_epoch_tags_single_string(self):
+        tags1 = 't1'
+        tags2 = 't2'
+        expected_tags = set([tags1, tags2])
+        self.nwbfile.add_epoch(0.0, 1.0, tags=tags1)
+        self.nwbfile.add_epoch(1.0, 2.0, tags=tags2)
+        tags = self.nwbfile.epoch_tags
+        self.assertEqual(expected_tags, tags)
+
     def test_add_acquisition(self):
         self.nwbfile.add_acquisition(TimeSeries('test_ts', [0, 1, 2, 3, 4, 5],
                                                 'grams', timestamps=[0.0, 0.1, 0.2, 0.3, 0.4, 0.5]))