Skip to content

Commit b7914b6

Browse files
author
Jan Sebastian Rothe
committed
attempt at filepath fix
1 parent c96c1f4 commit b7914b6

File tree

1 file changed

+24
-3
lines changed

1 file changed

+24
-3
lines changed

tests/paths/test_versions.py

Lines changed: 24 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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")
5371
def 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

Comments
 (0)