From c4234800a05e7df079a6266119b9e4cf2a92aca9 Mon Sep 17 00:00:00 2001 From: Yay295 Date: Fri, 1 Mar 2024 08:47:50 -0600 Subject: [PATCH] parametrize test_seek_mode functions --- Tests/test_file_container.py | 41 +++++++++--------------------------- 1 file changed, 10 insertions(+), 31 deletions(-) diff --git a/Tests/test_file_container.py b/Tests/test_file_container.py index 813b444db..1c1f58500 100644 --- a/Tests/test_file_container.py +++ b/Tests/test_file_container.py @@ -21,9 +21,16 @@ def test_isatty() -> None: assert container.isatty() is False -def test_seek_mode_0() -> None: +@pytest.mark.parametrize( + "mode, expected_value", + ( + (0, 33), + (1, 66), + (2, 100), + ), +) +def test_seek_mode(mode: int, expected_value: int) -> None: # Arrange - mode = 0 with open(TEST_FILE, "rb") as fh: container = ContainerIO.ContainerIO(fh, 22, 100) @@ -32,35 +39,7 @@ def test_seek_mode_0() -> None: container.seek(33, mode) # Assert - assert container.tell() == 33 - - -def test_seek_mode_1() -> None: - # Arrange - mode = 1 - with open(TEST_FILE, "rb") as fh: - container = ContainerIO.ContainerIO(fh, 22, 100) - - # Act - container.seek(33, mode) - container.seek(33, mode) - - # Assert - assert container.tell() == 66 - - -def test_seek_mode_2() -> None: - # Arrange - mode = 2 - with open(TEST_FILE, "rb") as fh: - container = ContainerIO.ContainerIO(fh, 22, 100) - - # Act - container.seek(33, mode) - container.seek(33, mode) - - # Assert - assert container.tell() == 100 + assert container.tell() == expected_value @pytest.mark.parametrize("bytesmode", (True, False))