fix: test patches updated to reflect code refactoring

tests/test_obs_map.py

@@ -9,10 +9,11 @@ from dataset.download import try_download_dataset
 #   is the return value, which should have similar form but change according to if an exception was raised or not
 # since this function uses st and m_logger to keep track of the download status, we need to mock them too
 
-@patch('maps.obs_map.load_dataset')
-@patch('maps.obs_map.st')
-@patch('maps.obs_map.m_logger')
-def test_try_download_dataset_success(mock_logger, mock_st, mock_load_dataset):
+#@patch('maps.obs_map.load_dataset')
+#@patch('maps.obs_map.m_logger')
+@patch('dataset.download.load_dataset')
+@patch('dataset.download.m_logger')
+def test_try_download_dataset_success(mock_logger, mock_load_dataset):
     # Mock the return value of load_dataset
     mock_load_dataset.return_value = {'train': {'latitude': [1], 'longitude': [2], 'predicted_class': ['whale']}}
 
@@ -25,13 +26,11 @@ def test_try_download_dataset_success(mock_logger, mock_st, mock_load_dataset):
     mock_load_dataset.assert_called_once_with(dataset_id, data_files=data_files)
     assert result == {'train': {'latitude': [1], 'longitude': [2], 'predicted_class': ['whale']}}
     mock_logger.info.assert_called_with("Downloaded dataset: (after 0.00s). ")
-    mock_st.write.assert_called_with("Downloaded dataset: (after 0.00s). ")
 
 
-@patch('maps.obs_map.load_dataset', side_effect=ValueError("Download failed"))
-@patch('maps.obs_map.st')
-@patch('maps.obs_map.m_logger')
-def test_try_download_dataset_failure_known(mock_logger, mock_st, mock_load_dataset):
+@patch('dataset.download.load_dataset', side_effect=ValueError("Download failed"))
+@patch('dataset.download.m_logger')
+def test_try_download_dataset_failure_known(mock_logger, mock_load_dataset):
     # testing the case where we've found (can reproduce by removing network connection)
     dataset_id = "test_dataset"
     data_files = "test_file"
@@ -41,15 +40,12 @@ def test_try_download_dataset_failure_known(mock_logger, mock_st, mock_load_data
     mock_logger.info.assert_any_call(f"Starting to download dataset {dataset_id} from Hugging Face")
     mock_load_dataset.assert_called_once_with(dataset_id, data_files=data_files)
     mock_logger.error.assert_called_with("Error downloading dataset: Download failed.  (after 0.00s).")
-    mock_st.error.assert_called_with("Error downloading dataset: Download failed.  (after 0.00s).")
     assert result == {}
     mock_logger.info.assert_called_with("Downloaded dataset: (after 0.00s). ")
-    mock_st.write.assert_called_with("Downloaded dataset: (after 0.00s). ")
 
-@patch('maps.obs_map.load_dataset', side_effect=Exception("Download engine corrupt"))
-@patch('maps.obs_map.st')
-@patch('maps.obs_map.m_logger')
-def test_try_download_dataset_failure_unknown(mock_logger, mock_st, mock_load_dataset):
+@patch('dataset.download.load_dataset', side_effect=Exception("Download engine corrupt"))
+@patch('dataset.download.m_logger')
+def test_try_download_dataset_failure_unknown(mock_logger, mock_load_dataset):
     # the cases we haven't found, but should still be handled (maybe network error, etc)
     dataset_id = "test_dataset"
     data_files = "test_file"
@@ -59,7 +55,5 @@ def test_try_download_dataset_failure_unknown(mock_logger, mock_st, mock_load_da
     mock_logger.info.assert_any_call(f"Starting to download dataset {dataset_id} from Hugging Face")
     mock_load_dataset.assert_called_once_with(dataset_id, data_files=data_files)
     mock_logger.error.assert_called_with("!!Unknown Error!! downloading dataset: Download engine corrupt.  (after 0.00s).")
-    mock_st.error.assert_called_with("!!Unknown Error!! downloading dataset: Download engine corrupt.  (after 0.00s).")
     assert result == {}
     mock_logger.info.assert_called_with("Downloaded dataset: (after 0.00s). ")
-    mock_st.write.assert_called_with("Downloaded dataset: (after 0.00s). ")