cp-library
This documentation is automatically generated by online-judge-tools/verification-helper
test/unittests/io/io_cls_test.py
- View this file on GitHub
- Last update: 2025-08-24 20:27:05+09:00
- Problem: https://onlinejudge.u-aizu.ac.jp/courses/lesson/2/ITP1/1/ITP1_1_A
Depends on
- cp_library/alg/dp/max2_fn.py
- cp_library/io/io_base_cls.py
- cp_library/io/io_bytes_cls.py
- cp_library/io/io_cls.py
- cp_library/test/unittest_helper.py
Code
# verification-helper: PROBLEM https://onlinejudge.u-aizu.ac.jp/courses/lesson/2/ITP1/1/ITP1_1_A
import pytest
import io
class TestIOBytes:
def test_initialization(self):
"""Test basic initialization of IO class"""
buffer = io.BytesIO(b"test\n")
test_io = IOBytes(buffer)
assert test_io.char == False
assert test_io.l == 0
assert test_io.p == 0
assert test_io.st == []
assert test_io.ist == []
def test_readtoken(self):
"""Test readtoken method"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
assert test_io.readtoken() == "hello"
assert test_io.l == 0 and test_io.p == 6
assert test_io.readtoken() == "world"
assert test_io.l == 0 and test_io.p == 12
assert test_io.readtoken() == "test"
assert test_io.l == 1 and test_io.p == 17
def test_readtokens(self):
"""Test readtokens method"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
tokens = test_io.readtokens()
assert tokens == ["hello", "world", "test"]
assert test_io.l == 1 and test_io.p == 17
def test_readints(self):
"""Test readints method"""
buffer = io.BytesIO(b"10 -20 300 -4000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [10, -20, 300, -4000]
assert test_io.l == 1 and test_io.p == 17
def test_readints_single_line(self):
"""Test readints with various integer formats"""
buffer = io.BytesIO(b"0 1 -1 42 -999 1000000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [0, 1, -1, 42, -999, 1000000]
assert test_io.l == 1 and test_io.p == 23
def test_readdigits_char_mode(self):
"""Test readdigits method in char mode"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
test_io.char = True
digits = test_io.readdigits()
assert digits == [1, 2, 3, 4, 5]
assert test_io.l == 1 and test_io.p == 6
def test_readnums_token_mode(self):
"""Test readnums in token mode (should use readints)"""
buffer = io.BytesIO(b"10 -20 300\n")
test_io = IOBytes(buffer)
test_io.char = False
nums = test_io.readnums()
assert nums == [10, -20, 300]
def test_readnums_char_mode(self):
"""Test readnums in char mode (should use readdigits)"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
test_io.char = True
nums = test_io.readnums()
assert nums == [1, 2, 3, 4, 5]
def test_readchar(self):
"""Test readchar method"""
buffer = io.BytesIO(b"abc\n")
test_io = IOBytes(buffer)
test_io.char = True
assert test_io.readchar() == "a"
assert test_io.l == 0 and test_io.p == 1
assert test_io.readchar() == "b"
assert test_io.l == 0 and test_io.p == 2
assert test_io.readchar() == "c"
assert test_io.l == 0 and test_io.p == 3
def test_readchars(self):
"""Test readchars method"""
buffer = io.BytesIO(b"hello\n")
test_io = IOBytes(buffer)
test_io.char = True
chars = test_io.readchars()
assert chars == "hello"
assert test_io.l == 1 and test_io.p == 6
def test_readline(self):
"""Test readline method"""
buffer = io.BytesIO(b"first line\nsecond line\n")
test_io = IOBytes(buffer)
assert test_io.readline() == "first line\n"
assert test_io.l == 1 and test_io.p == 11
assert test_io.readline() == "second line\n"
assert test_io.l == 2 and test_io.p == 23
def test_readinto_token_mode(self):
"""Test readinto in token mode"""
buffer = io.BytesIO(b"hello world\n")
test_io = IOBytes(buffer)
test_io.char = False
lst = []
result = test_io.readinto(lst)
assert result == ["hello", "world"]
assert lst == ["hello", "world"]
def test_readinto_char_mode(self):
"""Test readinto in char mode"""
buffer = io.BytesIO(b"hello\n")
test_io = IOBytes(buffer)
test_io.char = True
lst = []
result = test_io.readinto(lst)
assert "".join(result) == "hello"
assert "".join(lst) == "hello"
assert result == ['h', 'e', 'l', 'l', 'o']
assert lst == ['h', 'e', 'l', 'l', 'o']
def test_readtokensinto(self):
"""Test readtokensinto method"""
buffer = io.BytesIO(b"one two three\n")
test_io = IOBytes(buffer)
lst = ["existing"]
result = test_io.readtokensinto(lst)
assert result == ["existing", "one", "two", "three"]
assert lst == ["existing", "one", "two", "three"]
def test_readintsinto(self):
"""Test readintsinto method"""
buffer = io.BytesIO(b"10 -20 30\n")
test_io = IOBytes(buffer)
lst = [99]
result = test_io.readintsinto(lst)
assert result == [99, 10, -20, 30]
assert lst == [99, 10, -20, 30]
def test_readdigitsinto(self):
"""Test readdigitsinto method"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
lst = [9]
result = test_io.readdigitsinto(lst)
assert result == [9, 1, 2, 3, 4, 5]
assert lst == [9, 1, 2, 3, 4, 5]
def test_readnumsinto_token_mode(self):
"""Test readnumsinto in token mode"""
buffer = io.BytesIO(b"10 -20 30\n")
test_io = IOBytes(buffer)
test_io.char = False
lst = []
result = test_io.readnumsinto(lst)
assert result == [10, -20, 30]
assert lst == [10, -20, 30]
def test_readnumsinto_char_mode(self):
"""Test readnumsinto in char mode"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
test_io.char = True
lst = []
result = test_io.readnumsinto(lst)
assert result == [1, 2, 3, 4, 5]
assert lst == [1, 2, 3, 4, 5]
def test_line_token_mode(self):
"""Test line method in token mode"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
test_io.char = False
line_data = test_io.line()
assert line_data == ["hello", "world", "test"]
def test_line_char_mode(self):
"""Test line method in char mode"""
buffer = io.BytesIO(b"hello\n")
test_io = IOBytes(buffer)
test_io.char = True
line_data = test_io.line()
assert "".join(line_data) == "hello"
assert line_data == ['h', 'e', 'l', 'l', 'o']
assert len(line_data) == 5
def test_next_token_mode(self):
"""Test __next__ method in token mode"""
buffer = io.BytesIO(b"hello world\n")
test_io = IOBytes(buffer)
test_io.char = False
assert next(test_io) == "hello"
assert next(test_io) == "world"
def test_next_char_mode(self):
"""Test __next__ method in char mode"""
buffer = io.BytesIO(b"abc\n")
test_io = IOBytes(buffer)
test_io.char = True
assert next(test_io) == "a"
assert next(test_io) == "b"
assert next(test_io) == "c"
def test_multiline_tokens(self):
"""Test reading tokens across multiple lines"""
buffer = io.BytesIO(b"line1 data\nline2 more\n")
test_io = IOBytes(buffer)
# First line
tokens1 = test_io.readtokens()
assert tokens1 == ["line1", "data"]
assert test_io.l == 1 and test_io.p == 11
# Second line
tokens2 = test_io.readtokens()
assert tokens2 == ["line2", "more"]
assert test_io.l == 2 and test_io.p == 22
def test_multiline_ints(self):
"""Test reading integers across multiple lines"""
buffer = io.BytesIO(b"10 20\n30 40\n")
test_io = IOBytes(buffer)
# First line
ints1 = test_io.readints()
assert ints1 == [10, 20]
assert test_io.l == 1 and test_io.p == 6
# Second line
ints2 = test_io.readints()
assert ints2 == [30, 40]
assert test_io.l == 2 and test_io.p == 12
def test_empty_line(self):
"""Test handling empty lines"""
buffer = io.BytesIO(b"\n")
test_io = IOBytes(buffer)
tokens = test_io.readtokens()
assert tokens == [""]
assert test_io.l == 1 and test_io.p == 1
def test_single_integer(self):
"""Test reading single integer"""
buffer = io.BytesIO(b"42\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [42]
assert test_io.l == 1 and test_io.p == 3
def test_single_digit(self):
"""Test reading single digit in char mode"""
buffer = io.BytesIO(b"7\n")
test_io = IOBytes(buffer)
test_io.char = True
digits = test_io.readdigits()
assert digits == [7]
assert test_io.l == 1 and test_io.p == 2
def test_zero_handling(self):
"""Test proper handling of zero values"""
buffer = io.BytesIO(b"0 00 000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [0, 0, 0]
assert test_io.l == 1 and test_io.p == 9
def test_negative_zero(self):
"""Test handling of negative zero"""
buffer = io.BytesIO(b"-0\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [0]
assert test_io.l == 1 and test_io.p == 3
def test_large_numbers(self):
"""Test handling of large numbers"""
buffer = io.BytesIO(b"1000000000 -1000000000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [1000000000, -1000000000]
assert test_io.l == 1 and test_io.p == 23
def test_digits_with_linebreak(self):
"""Test digits reading stops at linebreak and advances line"""
buffer = io.BytesIO(b"123\n456\n")
test_io = IOBytes(buffer)
test_io.char = True
# First line digits
digits1 = test_io.readdigits()
assert digits1 == [1, 2, 3]
# Second line digits
digits2 = test_io.readdigits()
assert digits2 == [4, 5, 6]
def test_mixed_whitespace(self):
"""Test handling various whitespace characters"""
buffer = io.BytesIO(b"10 20 30\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [10, 20, 30]
assert test_io.l == 1 and test_io.p == 12
def test_char_mode_individual_access(self):
"""Test individual character access in char mode"""
buffer = io.BytesIO(b"abc123\n")
test_io = IOBytes(buffer)
test_io.char = True
line_data = test_io.line()
assert line_data[0] == 'a'
assert line_data[1] == 'b'
assert line_data[2] == 'c'
assert line_data[3] == '1'
assert line_data[4] == '2'
assert line_data[5] == '3'
assert len(line_data) == 6
def test_char_mode_readcharsinto_individual(self):
"""Test readcharsinto individual character behavior"""
buffer = io.BytesIO(b"test\n")
test_io = IOBytes(buffer)
lst = ['start']
result = test_io.readcharsinto(lst)
# readcharsinto extends with string characters
assert lst == ['start', 't', 'e', 's', 't']
assert result == ['start', 't', 'e', 's', 't']
def test_position_tracking_tokens(self):
"""Test io.p and io.l position tracking with single space separated tokens"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
# Initial state
assert test_io.l == 0
assert test_io.p == 0
# After first token
token1 = test_io.readtoken()
assert token1 == "hello"
assert test_io.l == 0 # Still on same line
assert test_io.p == 6 # Position after "hello "
# After second token
token2 = test_io.readtoken()
assert token2 == "world"
assert test_io.l == 0 # Still on same line
assert test_io.p == 12 # Position after "world "
# After third token (end of line)
token3 = test_io.readtoken()
assert token3 == "test"
assert test_io.l == 1 # Advanced to next line
assert test_io.p == 17 # Position at start of next line
def test_position_tracking_multiline(self):
"""Test io.p and io.l tracking across multiple lines"""
buffer = io.BytesIO(b"a b\nc d\n")
test_io = IOBytes(buffer)
# Line 0
assert test_io.readtoken() == "a"
assert test_io.l == 0 and test_io.p == 2
assert test_io.readtoken() == "b"
assert test_io.l == 1 and test_io.p == 4 # Next line start
# Line 1
assert test_io.readtoken() == "c"
assert test_io.l == 1 and test_io.p == 6
assert test_io.readtoken() == "d"
assert test_io.l == 2 and test_io.p == 8 # Next line start
def test_position_tracking_integers(self):
"""Test position tracking with integer reading"""
buffer = io.BytesIO(b"10 -20 300\n")
test_io = IOBytes(buffer)
# Read all integers at once
ints = test_io.readints()
assert ints == [10, -20, 300]
assert test_io.l == 1 # Advanced to next line
assert test_io.p == 11 # Position at start of next line
def test_position_tracking_char_mode(self):
"""Test position tracking in char mode"""
buffer = io.BytesIO(b"abc\ndef\n")
test_io = IOBytes(buffer)
test_io.char = True
# Read first character
assert test_io.readchar() == "a"
assert test_io.l == 0 and test_io.p == 1
# Read second character
assert test_io.readchar() == "b"
assert test_io.l == 0 and test_io.p == 2
# Read third character
assert test_io.readchar() == "c"
assert test_io.l == 0 and test_io.p == 3
# Read newline - should advance to next line
assert test_io.readchar() == "\n"
assert test_io.l == 1 and test_io.p == 4
def test_position_tracking_digits(self):
"""Test position tracking with digit reading in char mode"""
buffer = io.BytesIO(b"123\n456\n")
test_io = IOBytes(buffer)
test_io.char = True
# Read first line digits
digits1 = test_io.readdigits()
assert digits1 == [1, 2, 3]
assert test_io.l == 1 # Advanced to next line
assert test_io.p == 4 # Position at start of next line
# Read second line digits
digits2 = test_io.readdigits()
assert digits2 == [4, 5, 6]
assert test_io.l == 2 # Advanced to next line
assert test_io.p == 8 # Position at start of next line
def test_position_single_token_per_line(self):
"""Test position tracking with one token per line"""
buffer = io.BytesIO(b"first\nsecond\nthird\n")
test_io = IOBytes(buffer)
assert test_io.readtoken() == "first"
assert test_io.l == 1 and test_io.p == 6
assert test_io.readtoken() == "second"
assert test_io.l == 2 and test_io.p == 13
assert test_io.readtoken() == "third"
assert test_io.l == 3 and test_io.p == 19
def test_position_mixed_tokens_and_lines(self):
"""Test position tracking with mixed token patterns"""
buffer = io.BytesIO(b"1 2\n3\n4 5 6\n")
test_io = IOBytes(buffer)
# Line 0: "1 2"
tokens1 = test_io.readtokens()
assert tokens1 == ["1", "2"]
assert test_io.l == 1 and test_io.p == 4
# Line 1: "3"
tokens2 = test_io.readtokens()
assert tokens2 == ["3"]
assert test_io.l == 2 and test_io.p == 6
# Line 2: "4 5 6"
tokens3 = test_io.readtokens()
assert tokens3 == ["4", "5", "6"]
assert test_io.l == 3 and test_io.p == 12
from cp_library.io.io_bytes_cls import IOBytes
if __name__ == '__main__':
from cp_library.test.unittest_helper import run_verification_helper_unittest
run_verification_helper_unittest()# verification-helper: PROBLEM https://onlinejudge.u-aizu.ac.jp/courses/lesson/2/ITP1/1/ITP1_1_A
import pytest
import io
class TestIOBytes:
def test_initialization(self):
"""Test basic initialization of IO class"""
buffer = io.BytesIO(b"test\n")
test_io = IOBytes(buffer)
assert test_io.char == False
assert test_io.l == 0
assert test_io.p == 0
assert test_io.st == []
assert test_io.ist == []
def test_readtoken(self):
"""Test readtoken method"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
assert test_io.readtoken() == "hello"
assert test_io.l == 0 and test_io.p == 6
assert test_io.readtoken() == "world"
assert test_io.l == 0 and test_io.p == 12
assert test_io.readtoken() == "test"
assert test_io.l == 1 and test_io.p == 17
def test_readtokens(self):
"""Test readtokens method"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
tokens = test_io.readtokens()
assert tokens == ["hello", "world", "test"]
assert test_io.l == 1 and test_io.p == 17
def test_readints(self):
"""Test readints method"""
buffer = io.BytesIO(b"10 -20 300 -4000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [10, -20, 300, -4000]
assert test_io.l == 1 and test_io.p == 17
def test_readints_single_line(self):
"""Test readints with various integer formats"""
buffer = io.BytesIO(b"0 1 -1 42 -999 1000000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [0, 1, -1, 42, -999, 1000000]
assert test_io.l == 1 and test_io.p == 23
def test_readdigits_char_mode(self):
"""Test readdigits method in char mode"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
test_io.char = True
digits = test_io.readdigits()
assert digits == [1, 2, 3, 4, 5]
assert test_io.l == 1 and test_io.p == 6
def test_readnums_token_mode(self):
"""Test readnums in token mode (should use readints)"""
buffer = io.BytesIO(b"10 -20 300\n")
test_io = IOBytes(buffer)
test_io.char = False
nums = test_io.readnums()
assert nums == [10, -20, 300]
def test_readnums_char_mode(self):
"""Test readnums in char mode (should use readdigits)"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
test_io.char = True
nums = test_io.readnums()
assert nums == [1, 2, 3, 4, 5]
def test_readchar(self):
"""Test readchar method"""
buffer = io.BytesIO(b"abc\n")
test_io = IOBytes(buffer)
test_io.char = True
assert test_io.readchar() == "a"
assert test_io.l == 0 and test_io.p == 1
assert test_io.readchar() == "b"
assert test_io.l == 0 and test_io.p == 2
assert test_io.readchar() == "c"
assert test_io.l == 0 and test_io.p == 3
def test_readchars(self):
"""Test readchars method"""
buffer = io.BytesIO(b"hello\n")
test_io = IOBytes(buffer)
test_io.char = True
chars = test_io.readchars()
assert chars == "hello"
assert test_io.l == 1 and test_io.p == 6
def test_readline(self):
"""Test readline method"""
buffer = io.BytesIO(b"first line\nsecond line\n")
test_io = IOBytes(buffer)
assert test_io.readline() == "first line\n"
assert test_io.l == 1 and test_io.p == 11
assert test_io.readline() == "second line\n"
assert test_io.l == 2 and test_io.p == 23
def test_readinto_token_mode(self):
"""Test readinto in token mode"""
buffer = io.BytesIO(b"hello world\n")
test_io = IOBytes(buffer)
test_io.char = False
lst = []
result = test_io.readinto(lst)
assert result == ["hello", "world"]
assert lst == ["hello", "world"]
def test_readinto_char_mode(self):
"""Test readinto in char mode"""
buffer = io.BytesIO(b"hello\n")
test_io = IOBytes(buffer)
test_io.char = True
lst = []
result = test_io.readinto(lst)
assert "".join(result) == "hello"
assert "".join(lst) == "hello"
assert result == ['h', 'e', 'l', 'l', 'o']
assert lst == ['h', 'e', 'l', 'l', 'o']
def test_readtokensinto(self):
"""Test readtokensinto method"""
buffer = io.BytesIO(b"one two three\n")
test_io = IOBytes(buffer)
lst = ["existing"]
result = test_io.readtokensinto(lst)
assert result == ["existing", "one", "two", "three"]
assert lst == ["existing", "one", "two", "three"]
def test_readintsinto(self):
"""Test readintsinto method"""
buffer = io.BytesIO(b"10 -20 30\n")
test_io = IOBytes(buffer)
lst = [99]
result = test_io.readintsinto(lst)
assert result == [99, 10, -20, 30]
assert lst == [99, 10, -20, 30]
def test_readdigitsinto(self):
"""Test readdigitsinto method"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
lst = [9]
result = test_io.readdigitsinto(lst)
assert result == [9, 1, 2, 3, 4, 5]
assert lst == [9, 1, 2, 3, 4, 5]
def test_readnumsinto_token_mode(self):
"""Test readnumsinto in token mode"""
buffer = io.BytesIO(b"10 -20 30\n")
test_io = IOBytes(buffer)
test_io.char = False
lst = []
result = test_io.readnumsinto(lst)
assert result == [10, -20, 30]
assert lst == [10, -20, 30]
def test_readnumsinto_char_mode(self):
"""Test readnumsinto in char mode"""
buffer = io.BytesIO(b"12345\n")
test_io = IOBytes(buffer)
test_io.char = True
lst = []
result = test_io.readnumsinto(lst)
assert result == [1, 2, 3, 4, 5]
assert lst == [1, 2, 3, 4, 5]
def test_line_token_mode(self):
"""Test line method in token mode"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
test_io.char = False
line_data = test_io.line()
assert line_data == ["hello", "world", "test"]
def test_line_char_mode(self):
"""Test line method in char mode"""
buffer = io.BytesIO(b"hello\n")
test_io = IOBytes(buffer)
test_io.char = True
line_data = test_io.line()
assert "".join(line_data) == "hello"
assert line_data == ['h', 'e', 'l', 'l', 'o']
assert len(line_data) == 5
def test_next_token_mode(self):
"""Test __next__ method in token mode"""
buffer = io.BytesIO(b"hello world\n")
test_io = IOBytes(buffer)
test_io.char = False
assert next(test_io) == "hello"
assert next(test_io) == "world"
def test_next_char_mode(self):
"""Test __next__ method in char mode"""
buffer = io.BytesIO(b"abc\n")
test_io = IOBytes(buffer)
test_io.char = True
assert next(test_io) == "a"
assert next(test_io) == "b"
assert next(test_io) == "c"
def test_multiline_tokens(self):
"""Test reading tokens across multiple lines"""
buffer = io.BytesIO(b"line1 data\nline2 more\n")
test_io = IOBytes(buffer)
# First line
tokens1 = test_io.readtokens()
assert tokens1 == ["line1", "data"]
assert test_io.l == 1 and test_io.p == 11
# Second line
tokens2 = test_io.readtokens()
assert tokens2 == ["line2", "more"]
assert test_io.l == 2 and test_io.p == 22
def test_multiline_ints(self):
"""Test reading integers across multiple lines"""
buffer = io.BytesIO(b"10 20\n30 40\n")
test_io = IOBytes(buffer)
# First line
ints1 = test_io.readints()
assert ints1 == [10, 20]
assert test_io.l == 1 and test_io.p == 6
# Second line
ints2 = test_io.readints()
assert ints2 == [30, 40]
assert test_io.l == 2 and test_io.p == 12
def test_empty_line(self):
"""Test handling empty lines"""
buffer = io.BytesIO(b"\n")
test_io = IOBytes(buffer)
tokens = test_io.readtokens()
assert tokens == [""]
assert test_io.l == 1 and test_io.p == 1
def test_single_integer(self):
"""Test reading single integer"""
buffer = io.BytesIO(b"42\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [42]
assert test_io.l == 1 and test_io.p == 3
def test_single_digit(self):
"""Test reading single digit in char mode"""
buffer = io.BytesIO(b"7\n")
test_io = IOBytes(buffer)
test_io.char = True
digits = test_io.readdigits()
assert digits == [7]
assert test_io.l == 1 and test_io.p == 2
def test_zero_handling(self):
"""Test proper handling of zero values"""
buffer = io.BytesIO(b"0 00 000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [0, 0, 0]
assert test_io.l == 1 and test_io.p == 9
def test_negative_zero(self):
"""Test handling of negative zero"""
buffer = io.BytesIO(b"-0\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [0]
assert test_io.l == 1 and test_io.p == 3
def test_large_numbers(self):
"""Test handling of large numbers"""
buffer = io.BytesIO(b"1000000000 -1000000000\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [1000000000, -1000000000]
assert test_io.l == 1 and test_io.p == 23
def test_digits_with_linebreak(self):
"""Test digits reading stops at linebreak and advances line"""
buffer = io.BytesIO(b"123\n456\n")
test_io = IOBytes(buffer)
test_io.char = True
# First line digits
digits1 = test_io.readdigits()
assert digits1 == [1, 2, 3]
# Second line digits
digits2 = test_io.readdigits()
assert digits2 == [4, 5, 6]
def test_mixed_whitespace(self):
"""Test handling various whitespace characters"""
buffer = io.BytesIO(b"10 20 30\n")
test_io = IOBytes(buffer)
ints = test_io.readints()
assert ints == [10, 20, 30]
assert test_io.l == 1 and test_io.p == 12
def test_char_mode_individual_access(self):
"""Test individual character access in char mode"""
buffer = io.BytesIO(b"abc123\n")
test_io = IOBytes(buffer)
test_io.char = True
line_data = test_io.line()
assert line_data[0] == 'a'
assert line_data[1] == 'b'
assert line_data[2] == 'c'
assert line_data[3] == '1'
assert line_data[4] == '2'
assert line_data[5] == '3'
assert len(line_data) == 6
def test_char_mode_readcharsinto_individual(self):
"""Test readcharsinto individual character behavior"""
buffer = io.BytesIO(b"test\n")
test_io = IOBytes(buffer)
lst = ['start']
result = test_io.readcharsinto(lst)
# readcharsinto extends with string characters
assert lst == ['start', 't', 'e', 's', 't']
assert result == ['start', 't', 'e', 's', 't']
def test_position_tracking_tokens(self):
"""Test io.p and io.l position tracking with single space separated tokens"""
buffer = io.BytesIO(b"hello world test\n")
test_io = IOBytes(buffer)
# Initial state
assert test_io.l == 0
assert test_io.p == 0
# After first token
token1 = test_io.readtoken()
assert token1 == "hello"
assert test_io.l == 0 # Still on same line
assert test_io.p == 6 # Position after "hello "
# After second token
token2 = test_io.readtoken()
assert token2 == "world"
assert test_io.l == 0 # Still on same line
assert test_io.p == 12 # Position after "world "
# After third token (end of line)
token3 = test_io.readtoken()
assert token3 == "test"
assert test_io.l == 1 # Advanced to next line
assert test_io.p == 17 # Position at start of next line
def test_position_tracking_multiline(self):
"""Test io.p and io.l tracking across multiple lines"""
buffer = io.BytesIO(b"a b\nc d\n")
test_io = IOBytes(buffer)
# Line 0
assert test_io.readtoken() == "a"
assert test_io.l == 0 and test_io.p == 2
assert test_io.readtoken() == "b"
assert test_io.l == 1 and test_io.p == 4 # Next line start
# Line 1
assert test_io.readtoken() == "c"
assert test_io.l == 1 and test_io.p == 6
assert test_io.readtoken() == "d"
assert test_io.l == 2 and test_io.p == 8 # Next line start
def test_position_tracking_integers(self):
"""Test position tracking with integer reading"""
buffer = io.BytesIO(b"10 -20 300\n")
test_io = IOBytes(buffer)
# Read all integers at once
ints = test_io.readints()
assert ints == [10, -20, 300]
assert test_io.l == 1 # Advanced to next line
assert test_io.p == 11 # Position at start of next line
def test_position_tracking_char_mode(self):
"""Test position tracking in char mode"""
buffer = io.BytesIO(b"abc\ndef\n")
test_io = IOBytes(buffer)
test_io.char = True
# Read first character
assert test_io.readchar() == "a"
assert test_io.l == 0 and test_io.p == 1
# Read second character
assert test_io.readchar() == "b"
assert test_io.l == 0 and test_io.p == 2
# Read third character
assert test_io.readchar() == "c"
assert test_io.l == 0 and test_io.p == 3
# Read newline - should advance to next line
assert test_io.readchar() == "\n"
assert test_io.l == 1 and test_io.p == 4
def test_position_tracking_digits(self):
"""Test position tracking with digit reading in char mode"""
buffer = io.BytesIO(b"123\n456\n")
test_io = IOBytes(buffer)
test_io.char = True
# Read first line digits
digits1 = test_io.readdigits()
assert digits1 == [1, 2, 3]
assert test_io.l == 1 # Advanced to next line
assert test_io.p == 4 # Position at start of next line
# Read second line digits
digits2 = test_io.readdigits()
assert digits2 == [4, 5, 6]
assert test_io.l == 2 # Advanced to next line
assert test_io.p == 8 # Position at start of next line
def test_position_single_token_per_line(self):
"""Test position tracking with one token per line"""
buffer = io.BytesIO(b"first\nsecond\nthird\n")
test_io = IOBytes(buffer)
assert test_io.readtoken() == "first"
assert test_io.l == 1 and test_io.p == 6
assert test_io.readtoken() == "second"
assert test_io.l == 2 and test_io.p == 13
assert test_io.readtoken() == "third"
assert test_io.l == 3 and test_io.p == 19
def test_position_mixed_tokens_and_lines(self):
"""Test position tracking with mixed token patterns"""
buffer = io.BytesIO(b"1 2\n3\n4 5 6\n")
test_io = IOBytes(buffer)
# Line 0: "1 2"
tokens1 = test_io.readtokens()
assert tokens1 == ["1", "2"]
assert test_io.l == 1 and test_io.p == 4
# Line 1: "3"
tokens2 = test_io.readtokens()
assert tokens2 == ["3"]
assert test_io.l == 2 and test_io.p == 6
# Line 2: "4 5 6"
tokens3 = test_io.readtokens()
assert tokens3 == ["4", "5", "6"]
assert test_io.l == 3 and test_io.p == 12
'''
╺━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╸
https://kobejean.github.io/cp-library
'''
from __pypy__.builders import StringBuilder
from os import read as os_read, write as os_write, fstat as os_fstat
import sys
def max2(a, b): return a if a > b else b
class IOBase:
@property
def char(io) -> bool: ...
@property
def writable(io) -> bool: ...
def __next__(io) -> str: ...
def write(io, s: str) -> None: ...
def readline(io) -> str: ...
def readtoken(io) -> str: ...
def readtokens(io) -> list[str]: ...
def readints(io) -> list[int]: ...
def readdigits(io) -> list[int]: ...
def readnums(io) -> list[int]: ...
def readchar(io) -> str: ...
def readchars(io) -> str: ...
def readinto(io, lst: list[str]) -> list[str]: ...
def readcharsinto(io, lst: list[str]) -> list[str]: ...
def readtokensinto(io, lst: list[str]) -> list[str]: ...
def readintsinto(io, lst: list[int]) -> list[int]: ...
def readdigitsinto(io, lst: list[int]) -> list[int]: ...
def readnumsinto(io, lst: list[int]) -> list[int]: ...
def wait(io): ...
def flush(io) -> None: ...
def line(io) -> list[str]: ...
class IO(IOBase):
BUFSIZE = 1 << 16; stdin: 'IO'; stdout: 'IO'
__slots__ = 'f', 'file', 'B', 'O', 'V', 'S', 'l', 'p', 'char', 'sz', 'st', 'ist', 'writable', 'encoding', 'errors'
def __init__(io, file):
io.file = file
try: io.f = file.fileno(); io.sz, io.writable = max2(io.BUFSIZE, os_fstat(io.f).st_size), ('x' in file.mode or 'r' not in file.mode)
except: io.f, io.sz, io.writable = -1, io.BUFSIZE, False
io.B, io.O, io.S = bytearray(), [], StringBuilder(); io.V = memoryview(io.B); io.l = io.p = 0
io.char, io.st, io.ist, io.encoding, io.errors = False, [], [], 'ascii', 'ignore'
def _dec(io, l, r): return io.V[l:r].tobytes().decode(io.encoding, io.errors)
def readbytes(io, sz): return os_read(io.f, sz)
def load(io):
while io.l >= len(io.O):
if not (b := io.readbytes(io.sz)):
if io.O[-1] < len(io.B): io.O.append(len(io.B))
break
pos = len(io.B); io.B.extend(b)
while ~(pos := io.B.find(b'\n', pos)): io.O.append(pos := pos+1)
def __next__(io):
if io.char: return io.readchar()
else: return io.readtoken()
def readchar(io):
io.load(); r = io.O[io.l]
c = chr(io.B[io.p])
if io.p >= r-1: io.p = r; io.l += 1
else: io.p += 1
return c
def write(io, s: str): io.S.append(s)
def readline(io): io.load(); l, io.p = io.p, io.O[io.l]; io.l += 1; return io._dec(l, io.p)
def readtoken(io):
io.load(); r = io.O[io.l]
if ~(p := io.B.find(b' ', io.p, r)): s = io._dec(io.p, p); io.p = p+1
else: s = io._dec(io.p, r-1); io.p = r; io.l += 1
return s
def readtokens(io): io.st.clear(); return io.readtokensinto(io.st)
def readints(io): io.ist.clear(); return io.readintsinto(io.ist)
def readdigits(io): io.ist.clear(); return io.readdigitsinto(io.ist)
def readnums(io): io.ist.clear(); return io.readnumsinto(io.ist)
def readchars(io): io.load(); l, io.p = io.p, io.O[io.l]; io.l += 1; return io._dec(l, io.p-1)
def readinto(io, lst):
if io.char: return io.readcharsinto(lst)
else: return io.readtokensinto(lst)
def readcharsinto(io, lst): lst.extend(io.readchars()); return lst
def readtokensinto(io, lst):
io.load(); r = io.O[io.l]
while ~(p := io.B.find(b' ', io.p, r)): lst.append(io._dec(io.p, p)); io.p = p+1
lst.append(io._dec(io.p, r-1)); io.p = r; io.l += 1; return lst
def _readint(io, r):
while io.p < r and io.B[io.p] <= 32: io.p += 1
if io.p >= r: return None
minus = x = 0
if io.B[io.p] == 45: minus = 1; io.p += 1
while io.p < r and io.B[io.p] >= 48: x = x * 10 + (io.B[io.p] & 15); io.p += 1
io.p += 1
return -x if minus else x
def readintsinto(io, lst):
io.load(); r = io.O[io.l]
while io.p < r and (x := io._readint(r)) is not None: lst.append(x)
io.l += 1; return lst
def _readdigit(io): d = io.B[io.p] & 15; io.p += 1; return d
def readdigitsinto(io, lst):
io.load(); r = io.O[io.l]
while io.p < r and io.B[io.p] > 32: lst.append(io._readdigit())
if io.B[io.p] == 10: io.l += 1
io.p += 1
return lst
def readnumsinto(io, lst):
if io.char: return io.readdigitsinto(lst)
else: return io.readintsinto(lst)
def line(io): io.st.clear(); return io.readinto(io.st)
def wait(io):
io.load(); r = io.O[io.l]
while io.p < r: yield
def flush(io):
if io.writable: os_write(io.f, io.S.build().encode(io.encoding, io.errors)); io.S = StringBuilder()
sys.stdin = IO.stdin = IO(sys.stdin); sys.stdout = IO.stdout = IO(sys.stdout)
class IOBytes(IO):
def __init__(io, file):
io.file = file
io.f = -1
io.sz = io.BUFSIZE
io.B, io.O, io.S = bytearray(), [], StringBuilder()
io.V = memoryview(io.B)
io.l = io.p = 0
io.char = False
io.st, io.ist = [], []
io.writable, io.encoding, io.errors = True, 'ascii', 'ignore'
def readbytes(io, sz): return io.file.read(sz)
if __name__ == '__main__':
"""
Helper for making unittest files compatible with verification-helper.
This module provides a helper function to run a dummy Library Checker test
so that unittest files can be verified by oj-verify.
"""
def run_verification_helper_unittest():
"""
Run a dummy AOJ ITP1_1_A test for verification-helper compatibility.
This function should be called in the __main__ block of unittest files
that need to be compatible with verification-helper.
The function:
1. Prints "Hello World" (AOJ ITP1_1_A solution)
2. Runs pytest for the calling test file
3. Exits with the pytest result code
"""
# Print "Hello World" for AOJ ITP1_1_A problem
print("Hello World")
from contextlib import redirect_stdout, redirect_stderr
# Capture all output during test execution
output = io.StringIO()
with redirect_stdout(output), redirect_stderr(output):
# Get the calling module's file path
frame = sys._getframe(1)
test_file = frame.f_globals.get('__file__')
if test_file is None:
test_file = sys.argv[0]
result = pytest.main([test_file])
if result != 0:
print(output.getvalue())
sys.exit(result)
run_verification_helper_unittest()