@@ -34,8 +34,8 @@ def test_latest_version_path_gs(mock_get_fileversions, mock_latest_version_path)
3434 "gs://bucket/folder/file_v2.parquet" ,
3535 ]
3636 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"
37+ filepath = "gs://bucket/folder/file_v1.parquet"
38+ assert latest_version_path (filepath ) == "gs://bucket/folder/file_v2.parquet"
3939
4040
4141# Test for latest_version_path function local
@@ -47,8 +47,8 @@ def test_latest_version_path_local(mock_get_fileversions, mock_latest_version_pa
4747 "/local/folder/file_v2.parquet" ,
4848 ]
4949 mock_latest_version_path .return_value = "/local/folder/file_v2.parquet"
50- filepath_local = "/local/folder/file_v1.parquet"
51- assert latest_version_path (filepath_local ) == "/local/folder/file_v2.parquet"
50+ filepath = "/local/folder/file_v1.parquet"
51+ assert latest_version_path (filepath ) == "/local/folder/file_v2.parquet"
5252
5353
5454# Test for latest_version_number function gs
@@ -60,8 +60,8 @@ def test_latest_version_number_gs(mock_get_fileversions, mock_latest_version_pat
6060 "gs://bucket/folder/file_v2.parquet" ,
6161 ]
6262 mock_latest_version_path .return_value = "gs://bucket/folder/file_v2.parquet"
63- filepath_gs = "gs://bucket/folder/file_v1.parquet"
64- assert latest_version_number (filepath_gs ) == 2
63+ filepath = "gs://bucket/folder/file_v1.parquet"
64+ assert latest_version_number (filepath ) == 2
6565
6666
6767# Test for latest_version_number function local
@@ -73,8 +73,8 @@ def test_latest_version_number_local(mock_get_fileversions, mock_latest_version_
7373 "/local/folder/file_v2.parquet" ,
7474 ]
7575 mock_latest_version_path .return_value = "/local/folder/file_v2.parquet"
76- filepath_local = "/local/folder/file_v1.parquet"
77- assert latest_version_number (filepath_local ) == 2
76+ filepath = "/local/folder/file_v1.parquet"
77+ assert latest_version_number (filepath ) == 2
7878
7979
8080# Test for next_version_number function
@@ -86,8 +86,8 @@ def test_next_version_number(mock_get_fileversions, mock_latest_version_number):
8686 "gs://bucket/folder/file_v2.parquet" ,
8787 ]
8888 mock_latest_version_number .return_value = 2
89- filepath_gs = "gs://bucket/folder/file_v2.parquet"
90- assert next_version_number (filepath_gs ) == 3
89+ filepath = "gs://bucket/folder/file_v2.parquet"
90+ assert next_version_number (filepath ) == 3
9191
9292
9393# Test for next_version_path function
@@ -99,7 +99,7 @@ def test_next_version_path(
9999):
100100 mock_next_version_number .return_value = 3
101101 mock_latest_version_path .return_value = "gs://bucket/folder/file_v2.parquet"
102- mock_get_version_number .return_value = 1
102+ mock_get_version_number .return_value = 2
103103 file_path = "gs://bucket/folder/file_v1.parquet"
104104 expected = "gs://bucket/folder/file_v3.parquet"
105105 assert next_version_path (file_path ) == expected
0 commit comments