diff --git a/tests/common/hasher.py b/tests/common/hasher.py index 4da3ed6..4ea0118 100644 --- a/tests/common/hasher.py +++ b/tests/common/hasher.py @@ -13,7 +13,7 @@ def read_expected_hashes(which: str) -> pd.Series: fp = HASH_DIR / FILE_FMT.format(which=which) - return pd.read_csv(fp, index_col=[0, 1], squeeze=True) + return pd.read_csv(fp, index_col=[0, 1]).squeeze() def save_expected_hashes(which: str) -> None: diff --git a/tests/test_helper.py b/tests/test_helper.py index 32200cf..88f6393 100644 --- a/tests/test_helper.py +++ b/tests/test_helper.py @@ -68,7 +68,7 @@ def test_delete_cache(mocker, capsys): fp = paths.CACHE_DIR / "test_file_XXXX.parquet" tuplelist = [(True, "1 files deleted\n"), (False, "No files deleted\n")] - for (patch_value, msg) in tuplelist: + for patch_value, msg in tuplelist: fp.touch(exist_ok=True) assert fp.exists() mocker.patch("elmada.helper.confirm_deletion", return_value=patch_value) @@ -141,9 +141,9 @@ def test_read(mocker): for suffix, func in zip(suffices, funcs): fp = Path(f"_.{suffix}") mock = mocker.patch(func, return_value=pd.DataFrame({0: [1, 2, 3]})) - result = hp.read(fp, squeeze=True) + result = hp.read(fp).squeeze() assert isinstance(result, pd.Series) - result = hp.read(fp, squeeze=False) + result = hp.read(fp) assert isinstance(result, pd.DataFrame) assert mock.called