From ef436a6bf473fddeabf9b15f885e8d8f146a0ef9 Mon Sep 17 00:00:00 2001
From: Alex Spaeth <atspaeth@ucsc.edu>
Date: Wed, 22 May 2024 23:19:37 -0700
Subject: [PATCH] Fix path to test maxwell metadata

---
 tests/datasets_electrophysiology.py | 25 ++++++++++++-------------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/tests/datasets_electrophysiology.py b/tests/datasets_electrophysiology.py
index 807b576..116502e 100644
--- a/tests/datasets_electrophysiology.py
+++ b/tests/datasets_electrophysiology.py
@@ -1,18 +1,17 @@
-import unittest
-import tempfile
-import shutil
-import diskcache
 import json
+import shutil
+import tempfile
 import threading
-import braingeneers
+import unittest
+from unittest.mock import patch
+
+import diskcache
+import numpy as np
+
 import braingeneers.data.datasets_electrophysiology as ephys
-from braingeneers import skip_unittest_if_offline
 import braingeneers.utils.smart_open_braingeneers as smart_open
-import boto3
-import numpy as np
-from unittest.mock import patch
+from braingeneers import skip_unittest_if_offline
 from braingeneers.data.datasets_electrophysiology import cached_load_data
-from unittest.mock import patch
 
 
 class MaxwellReaderTests(unittest.TestCase):
@@ -113,17 +112,17 @@ def test_non_int_offset_length(self):
 
     def test_modify_maxwell_metadata(self):
         """Update an older Maxwell metadata json with new metadata and NWB file paths, if they exist."""
-        with open('test_data/maxwell-metadata.old.json', 'r') as f:
+        with open('tests/test_data/maxwell-metadata.old.json', 'r') as f:
             metadata = json.load(f)
             # use mock to ensure that new NWB files ALWAYS exist
-            with patch('__main__.s3wrangler.does_object_exist') as mock_does_object_exist:
+            with patch('braingeneers.utils.s3wrangler.does_object_exist') as mock_does_object_exist:
                 mock_does_object_exist.return_value = True
                 modified_metadata = ephys.modify_metadata_maxwell_raw_to_nwb(metadata)
         assert isinstance(modified_metadata['timestamp'], str)
         assert len(modified_metadata['timestamp']) == len('2023-10-05T18:10:02')
         modified_metadata['timestamp'] = ''
 
-        with open('test_data/maxwell-metadata.expected.json', 'r') as f:
+        with open('tests/test_data/maxwell-metadata.expected.json', 'r') as f:
             expected_metadata = json.load(f)
             expected_metadata['timestamp'] = ''