def test_example_input(self):
chart = SeatingChart(3, 11)
chart.initialize([(1, 4), (1, 6), (2, 3), (2, 7), (3, 9), (3, 10)])
chart._find_seats(3)
chart._find_seats(3)
chart._find_seats(3)
chart._find_seats(1)
chart._find_seats(10)
expected = {
300002: 1,
400001: 1,
400002: 1,
500001: 1,
500002: 1,
500003: 1,
600001: 1,
600002: 1,
600003: 1,
700001: 1,
700002: 1,
700003: 1,
800001: 1,
900001: 1,
900003: 1,
1000003: 1
}
actual = chart._reserved_seats
assert expected == actual
def test_find_best_five_seats_in_second_row(self):
chart = SeatingChart(5, 10)
chart.initialize([(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6),
(1, 7), (1, 8), (1, 9), (1, 10)])
chart._find_seats(5)
expected = {
400001: 1,
200001: 1,
100001: 1,
900001: 1,
700001: 1,
600001: 1,
500001: 1,
300001: 1,
1000001: 1,
800001: 1,
300002: 1,
400002: 1,
500002: 1,
600002: 1,
700002: 1
}
actual = chart._reserved_seats
assert expected == actual
def test_find_best_five_seats_in_empty_row_of_even_columns(self):
chart = SeatingChart(1, 10)
chart._find_seats(5)
expected = {300001: 1, 400001: 1, 500001: 1, 600001: 1, 700001: 1}
actual = chart._reserved_seats
assert expected == actual
def test_find_best_four_seats_in_empty_row_of_odd_columns(self):
chart = SeatingChart(1, 9)
chart._find_seats(4)
expected = {300001: 1, 400001: 1, 500001: 1, 600001: 1}
actual = chart._reserved_seats
assert expected == actual
def test_group_bigger_than_a_row(self):
chart = SeatingChart(10, 6)
expected = None
actual = chart._find_seats(7)
assert actual == expected
def test_no_space_left_for_group(self):
chart = SeatingChart(2, 6)
chart.initialize([(1, 1), (1, 3), (1, 5), (2, 2), (2, 4), (2, 6)])
expected = None
actual = chart._find_seats(2)
assert actual == expected
