@@ -25,6 +25,26 @@ def test_get_latest_fileversions():
2525 assert sorted (get_latest_fileversions (paths )) == sorted (expected )
2626
2727
28+ # Test for latest_version_path function
29+ @patch ("fagfunksjoner.paths.versions.get_fileversions" )
30+ @patch ("fagfunksjoner.paths.versions.latest_version_path" )
31+ def test_latest_version_path (mock_latest_version_path , mock_get_fileversions ):
32+ mock_get_fileversions .return_value = [
33+ "gs://bucket/folder/file_v1.parquet" ,
34+ "gs://bucket/folder/file_v2.parquet" ,
35+ ]
36+ mock_latest_version_path .return_value = "gs://bucket/folder/file_v2.parquet"
37+ filepath_gs = "gs://bucket/folder/file_v1.parquet"
38+ assert latest_version_path (filepath_gs ) == "gs://bucket/folder/file_v2.parquet"
39+ mock_get_fileversions .return_value = [
40+ "/local/folder/file_v1.parquet" ,
41+ "/local/folder/file_v2.parquet" ,
42+ ]
43+ mock_latest_version_path .return_value = "/local/folder/file_v2.parquet"
44+ filepath_local = "/local/folder/file_v1.parquet"
45+ assert latest_version_path (filepath_local ) == "/local/folder/file_v2.parquet"
46+
47+
2848# Test for latest_version_number function
2949@patch ("fagfunksjoner.paths.versions.get_fileversions" )
3050@patch ("fagfunksjoner.paths.versions.latest_version_path" )
@@ -36,22 +56,23 @@ def test_latest_version_number(mock_latest_version_path, mock_get_fileversions):
3656 mock_latest_version_path .return_value = "gs://bucket/folder/file_v2.parquet"
3757 filepath_gs = "gs://bucket/folder/file_v1.parquet"
3858 assert latest_version_number (filepath_gs ) == 2
39- assert latest_version_path (filepath_gs ) == "gs://bucket/folder/file_v2.parquet"
4059 mock_get_fileversions .return_value = [
4160 "/local/folder/file_v1.parquet" ,
4261 "/local/folder/file_v2.parquet" ,
4362 ]
4463 mock_latest_version_path .return_value = "/local/folder/file_v2.parquet"
4564 filepath_local = "/local/folder/file_v1.parquet"
4665 assert latest_version_number (filepath_local ) == 2
47- assert latest_version_path (filepath_local ) == "/local/folder/file_v2.parquet"
4866
4967
5068# Test for next_version_number function
5169@patch ("fagfunksjoner.paths.versions.get_fileversions" )
5270@patch ("fagfunksjoner.paths.versions.latest_version_path" )
5371def test_next_version_number (mock_get_fileversions , mock_latest_version_path ):
54- mock_get_fileversions .return_value = ["gs://bucket/folder/file_v2.parquet" ]
72+ mock_get_fileversions .return_value = [
73+ "gs://bucket/folder/file_v1.parquet" ,
74+ "gs://bucket/folder/file_v2.parquet" ,
75+ ]
5576 mock_latest_version_path .return_value = "gs://bucket/folder/file_v2.parquet"
5677 filepath_gs = "gs://bucket/folder/file_v2.parquet"
5778 assert next_version_number (filepath_gs ) == 3
0 commit comments