diff --git a/src/aind_metadata_mapper/FIB/session.py b/src/aind_metadata_mapper/FIB/session.py index a8a24185..87fdcdfd 100644 --- a/src/aind_metadata_mapper/FIB/session.py +++ b/src/aind_metadata_mapper/FIB/session.py @@ -82,7 +82,6 @@ def map_response_to_ophys_session( # maps stimulus_name from command command = command_match.group(1) stimulus_name = getattr(StimulusName, command, "") - print("stimulus", stimulus_name) # create opto stim instance opto_stim = OptoStimulation( diff --git a/tests/resources/FIB/000000_ophys_rig.json b/tests/resources/FIB/000000_ophys_rig.json index 0bed4622..131e90bc 100644 --- a/tests/resources/FIB/000000_ophys_rig.json +++ b/tests/resources/FIB/000000_ophys_rig.json @@ -443,7 +443,7 @@ "model": null, "path_to_cad": null, "notes": null, - "type": "Tunable lens" + "type": "Other" }, { "device_type": "AdditionalImagingDevice", @@ -458,7 +458,7 @@ "model": null, "path_to_cad": null, "notes": null, - "type": "Tunable lens" + "type": "Other" } ], "calibration_date": null, diff --git a/tests/test_FIB.py b/tests/test_FIB.py index 03b8fabb..352b5d86 100644 --- a/tests/test_FIB.py +++ b/tests/test_FIB.py @@ -87,17 +87,13 @@ def test_map_response_to_ophys_session(self): with open(ophys_session_path, "r") as f: actual_session_contents = json.load(f) - print( - actual_session_contents["stimulus_epochs"][0]["stimulus"][ - "stimulus_name" - ] - ) self.assertEqual( actual_session_contents["stimulus_epochs"][0]["stimulus"][ "stimulus_name" ], stimulus_name, ) + os.remove(ophys_session_path) def test_map_to_ophys_rig(self): """Tests that the teensy response maps correctly to ophys rig.""" @@ -110,17 +106,13 @@ def test_map_to_ophys_rig(self): reference_path=RESOURCES_DIR, ) - # ophys_rig_path = ( - # self.example_experiment_data["save_dir"] - # + f"/{self.example_experiment_data['labtracks_id']}_" - # + start_date.strftime("%Y-%m-%d_%H-%M-%S") - # + "_ophys_rig.json" - # ) - - # with open(ophys_rig_path, "r") as f: - # actual_rig_contents = json.load(f) - # - # self.assertEqual(actual_rig_contents, self.expected_rig) + ophys_rig_path = ( + self.example_experiment_data["save_dir"] + + f"/{self.example_experiment_data['labtracks_id']}_" + + start_date.strftime("%Y-%m-%d_%H-%M-%S") + + "_ophys_rig.json" + ) + os.remove(ophys_rig_path) if __name__ == "__main__":