adds unit tests

AllenNeuralDynamics · Sep 21, 2024 · ca25aa9 · ca25aa9
1 parent 57376c8
commit ca25aa9
Show file tree

Hide file tree

Showing 5 changed files with 2,356 additions and 6 deletions.
diff --git a/src/aind_slims_api/models/ecephys_result.py → src/aind_slims_api/models/ecephys_session.py b/src/aind_slims_api/models/ecephys_result.py → src/aind_slims_api/models/ecephys_session.py
@@ -1,5 +1,6 @@
 """Contains a model for an ecephys result stored in SLIMS."""
 
+from datetime import datetime
 from typing import Annotated, List, Optional, ClassVar
 from pydantic import Field
 from aind_slims_api.models.base import SlimsBaseModel
@@ -18,11 +19,6 @@ class SlimsStreamsResult(SlimsBaseModel):
         serialization_alias="rslt_cf_fk_mouseSession_display",
         validation_alias="rslt_cf_fk_mouseSession_display"
     )
-    mouse_session_pk: Optional[int] = Field(
-        default=None,
-        serialization_alias="rslt_cf_fk_mouseSession",
-        validation_alias="rslt_cf_fk_mouseSession"
-    )
     stream_modalities: Optional[List] = Field(
         default=None,
         serialization_alias="rslt_cf_streamModalities",
@@ -54,6 +50,16 @@ class SlimsStreamsResult(SlimsBaseModel):
         serialization_alias="rslt_fk_content",
         validation_alias="rslt_fk_content",
     )
+    mouse_session_pk: Optional[int] = Field(
+        default=None,
+        serialization_alias="rslt_cf_fk_mouseSession",
+        validation_alias="rslt_cf_fk_mouseSession"
+    )
+    created_on: Optional[datetime] = Field(
+        default=None,
+        serialization_alias="rslt_createdOn",
+        validation_alias="rslt_createdOn"
+    )
     _slims_table = "Result"
     _base_fetch_filters: ClassVar[dict[str, str]] = {
         "test_name": "test_ephys_in_vivo_recording_stream",
@@ -257,4 +263,3 @@ class SlimsEphysInsertionResult(SlimsBaseModel):
     _base_fetch_filters: ClassVar[dict[str, str]] = {
         "test_name": "test_ephys_insertion",
     }
-