@@ -25,17 +25,23 @@ def test_get_latest_fileversions():
2525 assert sorted (get_latest_fileversions (paths )) == sorted (expected )
2626
2727
28- # Test for latest_version_path function
28+ # Test for latest_version_path function gs
2929@patch ("fagfunksjoner.paths.versions.get_fileversions" )
3030@patch ("fagfunksjoner.paths.versions.latest_version_path" )
31- def test_latest_version_path ( mock_latest_version_path , mock_get_fileversions ):
31+ def test_latest_version_path_gs ( mock_get_fileversions , mock_latest_version_path ):
3232 mock_get_fileversions .return_value = [
3333 "gs://bucket/folder/file_v1.parquet" ,
3434 "gs://bucket/folder/file_v2.parquet" ,
3535 ]
3636 mock_latest_version_path .return_value = "gs://bucket/folder/file_v2.parquet"
3737 filepath_gs = "gs://bucket/folder/file_v1.parquet"
3838 assert latest_version_path (filepath_gs ) == "gs://bucket/folder/file_v2.parquet"
39+
40+
41+ # Test for latest_version_path function local
42+ @patch ("fagfunksjoner.paths.versions.get_fileversions" )
43+ @patch ("fagfunksjoner.paths.versions.latest_version_path" )
44+ def test_latest_version_path_local (mock_get_fileversions , mock_latest_version_path ):
3945 mock_get_fileversions .return_value = [
4046 "/local/folder/file_v1.parquet" ,
4147 "/local/folder/file_v2.parquet" ,
@@ -45,17 +51,23 @@ def test_latest_version_path(mock_latest_version_path, mock_get_fileversions):
4551 assert latest_version_path (filepath_local ) == "/local/folder/file_v2.parquet"
4652
4753
48- # Test for latest_version_number function
54+ # Test for latest_version_number function gs
4955@patch ("fagfunksjoner.paths.versions.get_fileversions" )
5056@patch ("fagfunksjoner.paths.versions.latest_version_path" )
51- def test_latest_version_number ( mock_latest_version_path , mock_get_fileversions ):
57+ def test_latest_version_number_gs ( mock_get_fileversions , mock_latest_version_path ):
5258 mock_get_fileversions .return_value = [
5359 "gs://bucket/folder/file_v1.parquet" ,
5460 "gs://bucket/folder/file_v2.parquet" ,
5561 ]
5662 mock_latest_version_path .return_value = "gs://bucket/folder/file_v2.parquet"
5763 filepath_gs = "gs://bucket/folder/file_v1.parquet"
5864 assert latest_version_number (filepath_gs ) == 2
65+
66+
67+ # Test for latest_version_number function local
68+ @patch ("fagfunksjoner.paths.versions.get_fileversions" )
69+ @patch ("fagfunksjoner.paths.versions.latest_version_path" )
70+ def test_latest_version_number_local (mock_get_fileversions , mock_latest_version_path ):
5971 mock_get_fileversions .return_value = [
6072 "/local/folder/file_v1.parquet" ,
6173 "/local/folder/file_v2.parquet" ,
0 commit comments