Skip to content

Commit 7d95cec

Browse files
author
Jan Sebastian Rothe
committed
attempt at test fix
1 parent 87bf991 commit 7d95cec

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

tests/paths/test_versions.py

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)