feat: add slap2 acquisition types (moved from aind-data-schema) (#270)

* feat: add slap2 acquisition types (moved from aind-data-schema)

* tests: coverage on new model

Author: dbirman

src/aind_data_schema_models/_generators/models/slap2_acquisition_type.csv

@@ -0,0 +1,5 @@
+name
+full-field raster scan
+multi-roi raster scan
+band scan
+integration scan

src/aind_data_schema_models/_generators/templates/slap2_acquisition_type.txt

@@ -0,0 +1,11 @@
+"""Stimulus modalities"""
+{% raw -%}
+from enum import Enum
+{% endraw %}
+
+class Slap2AcquisitionType(str, Enum):
+    """Stimulus modalities"""
+{% for _, row in data.iterrows() %}
+    {{ row['name'] | to_class_name | upper }} = "{{ row['name'] }}"
+{%- endfor %}
+

src/aind_data_schema_models/slap2_acquisition_type.py

@@ -0,0 +1,12 @@
+"""Stimulus modalities"""
+
+from enum import Enum
+
+
+class Slap2AcquisitionType(str, Enum):
+    """Stimulus modalities"""
+
+    BAND_SCAN = "band scan"
+    FULL_FIELD_RASTER_SCAN = "full-field raster scan"
+    INTEGRATION_SCAN = "integration scan"
+    MULTI_ROI_RASTER_SCAN = "multi-roi raster scan"

tests/test_slap2_acquisition_type.py

@@ -0,0 +1,18 @@
+"""Tests classes in slap2_acquisition_type module"""
+
+import unittest
+
+from aind_data_schema_models.slap2_acquisition_type import Slap2AcquisitionType
+
+
+class TestSlap2AcquisitionType(unittest.TestCase):
+    """Tests methods in Slap2AcquisitionType class"""
+
+    def test_class_construction(self):
+        """Tests enum can be instantiated via string"""
+
+        self.assertEqual(Slap2AcquisitionType.BAND_SCAN, Slap2AcquisitionType("band scan"))
+
+
+if __name__ == "__main__":
+    unittest.main()