Skip to content

Commit

Permalink
Merge pull request #7847 from Yay295/patch-1
Browse files Browse the repository at this point in the history
Parametrize test_seek_mode functions
  • Loading branch information
radarhere committed Mar 2, 2024
2 parents 334c26d + c423480 commit 2bd5426
Showing 1 changed file with 10 additions and 31 deletions.
41 changes: 10 additions & 31 deletions Tests/test_file_container.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand All @@ -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))
Expand Down

0 comments on commit 2bd5426

Please sign in to comment.