Update test_core.py

sarahshi · Jan 25, 2024 · 352ed0c · 352ed0c
1 parent b1f8ffb
commit 352ed0c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/UnitTests/test_core.py b/UnitTests/test_core.py
@@ -1,5 +1,6 @@
 import unittest
 from unittest.mock import patch, mock_open
+from tempfile import TemporaryFile
 
 import os
 import glob
@@ -117,8 +118,8 @@ def setUp(self):
                         'Mineral2', 'Mineral1', 'Mineral2', 'Mineral1', 'Mineral2']
         })
 
-    @patch('your_module.load_minclass_nn')
-    @patch('your_module.norm_data')
+    @patch('mm.load_minclass_nn')
+    @patch('mm.norm_data')
     def test_create_dataloader(self, mock_norm_data, mock_load_minclass_nn):
         # Mock return values for dependencies
         mock_norm_data.return_value = self.df