def test_encrypted_name():
r = Room('aaaaa-bbb-z-y-x-123[abxyz]')
assert r.encrypted_name == 'aaaaa-bbb-z-y-x'
def test_shift_string():
assert Room._shift_string('abc', 1) == 'bcd'
def test_real_name():
r = Room('qzmt-zixmtkozy-ivhz-343[dummy]')
assert r.real_name == 'very encrypted name'
def test_sector_id():
r = Room('aaaaa-bbb-z-y-x-123[abxyz]')
assert r.sector_id == 123
def test_shift_letter():
assert Room._shift_letter('a', 1) == 'b'
assert Room._shift_letter('z', 1) == 'a'
assert Room._shift_letter('a', 26) == 'a'
assert Room._shift_letter('a', 52) == 'a'
assert Room._shift_letter(' ', 1) == ' '
def test_code():
r = Room('aaaaa-bbb-z-y-x-123[abxyz]')
assert r.code == 'aaaaa-bbb-z-y-x-123[abxyz]'
def test_is_real():
r = Room('aaaaa-bbb-z-y-x-123[abxyz]')
assert r.is_real()
r = Room('a-b-c-d-e-f-g-h-987[abcde]')
assert r.is_real()
r = Room('not-a-real-room-404[oarel]')
assert r.is_real()
r = Room('totally-real-room-200[decoy]')
assert not r.is_real()
def test_calculate_checksum():
r = Room('aaaaa-bbb-z-y-x-123[abxyz]')
assert r.calculate_checksum() == 'abxyz'
def test_precalculated_checksum():
r = Room('aaaaa-bbb-z-y-x-123[abxyz]')
assert r.precalculated_checksum == 'abxyz'
r = Room('aaaaa-bbb-z-y-x-123[abxyz]\n')
assert r.precalculated_checksum == 'abxyz'
from day04 import Room
with open('input.txt') as f:
lines = f.readlines()
sum_ = 0
for line in lines:
r = Room(line)
if r.is_real():
sum_ += r.sector_id
if 'pole' in r.real_name:
print(r.real_name)
print(r.sector_id)
print(sum_)