def test_start_checkout() -> None:
"""Test the start_checkout function of the GroceryStore Class."""
gs = GroceryStore(StringIO(ONE_LINE_FILE_CONTENTS))
gs2 = GroceryStore(StringIO(SELF_SERVE_FILE_CONTENTS))
c1 = Customer('A', [Item('bananas', 7), Item('apple', 2), \
Item('orange', 5), Item('grapes', 10)])
c2 = Customer('B', [Item('apple', 2)])
c3 = Customer('C', [Item('orange', 5)])
gs.enter_line(c1)
gs.enter_line(c2)
gs2.enter_line(c3)
assert gs.start_checkout(0) == 24
assert gs2.start_checkout(0) == 10
def test_enter_line():
f = StringIO(CONFIG_FILE_000_00)
store1 = GroceryStore(f)
customer = Customer('Chris', [Item('bread', 3)])
assert store1.enter_line(customer) == -1
f.close()
f = StringIO(CONFIG_FILE_010_01)
store2 = GroceryStore(f)
lst = []
for i in range(EXPRESS_LIMIT + 2):
item = Item('pan', 2)
lst.append(item)
customer1 = Customer("Owen", lst)
assert store2.enter_line(customer1) == -1
def test_close_line_and_get_first_in_line() -> None:
"""Test the close_line function of the GroceryStore Class."""
gs = GroceryStore(StringIO(ONE_LINE_FILE_CONTENTS))
gs2 = GroceryStore(StringIO(SELF_SERVE_FILE_CONTENTS))
c1 = Customer('A', [Item('bananas', 7)])
c2 = Customer('B', [Item('apple', 2)])
c3 = Customer('C', [Item('orange', 5)])
c4 = Customer('D', [Item('grapes', 10)])
gs.enter_line(c1)
gs.enter_line(c2)
gs.enter_line(c3)
gs.enter_line(c4)
assert gs.close_line(0) == [c2, c3, c4]
assert gs.get_first_in_line(0) == c1
assert gs._checkout_lines[0].queue == [c1]
assert gs2.get_first_in_line(0) is None
def test_start_checkout_non_empty() -> None:
""" Test whether non empty line starts checkout."""
store = GroceryStore(GROCERY_STORE)
item_list = [Item('bananas', 6), Item('apples', 5), Item('carrots', 3)]
belinda = Customer('Belinda', item_list)
store.enter_line(belinda)
assert store.start_checkout(0) == 14
def test_close_line_one_customer() -> None:
""" Test whether line with one customer returns customers on close."""
store = GroceryStore(GROCERY_STORE)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
store.enter_line(belinda)
assert store.close_line(0) == []
def test_line_is_ready_one_occupant() -> None:
""" Test line is ready with exactly one customer."""
store = GroceryStore(GROCERY_STORE)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
store.enter_line(belinda)
assert store.line_is_ready(0)
def test_complete_checkout_one() -> None:
""" Test whether customers left to checkout after checkout when
line has one customer."""
store = GroceryStore(GROCERY_STORE)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
store.enter_line(belinda)
assert not store.complete_checkout(0)
def test_get_first_in_line_many() -> None:
""" Test whether get first in line returns correct customer with
one customer in line."""
store = GroceryStore(GROCERY_STORE)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
store.enter_line(belinda)
assert store.get_first_in_line(0) == belinda
def test_enter_line() -> None:
""" Test whether customer enters correct line with empty lines."""
store = GroceryStore(GROCERY_STORE)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
assert len(store._lines[0]) == 0
assert store.enter_line(belinda) == 0
assert len(store._lines[0]) == 1
def test_start_checkout_self_serve_line() -> None:
"""check for the checkout method for self serve line"""
store1 = GroceryStore(StringIO(CONFIG_FILE3))
item1 = Item("banana", 10)
item2 = Item("apple", 2)
item3 = Item("orange", 9)
customer1 = Customer("bruce", [item1, item2, item3])
assert store1.enter_line(customer1) == 0
assert store1.start_checkout(0) == 42
def test_line_is_ready() -> None:
"""check the method of line is ready"""
store1 = GroceryStore(StringIO(CONFIG_FILE2))
customer1 = Customer("bruce", [])
assert store1.line_is_ready(0) is False
assert store1.line_is_ready(1) is False
assert store1.enter_line(customer1) == 0
assert store1.line_is_ready(0) is True
assert store1.line_is_ready(1) is False
def __init__(self, store_file):
"""Initialize a GroceryStoreSimulation from a file.
@type store_file: str
A file containing the configuration of the grocery store.
@rtype: None
"""
self._events = PriorityQueue()
self._store = GroceryStore(store_file)
def test_line_is_ready() -> None:
"""Test the line_is_ready function of the GroceryStore Class."""
gs = GroceryStore(StringIO(SHORT_FILE_CONTENTS))
gs2 = GroceryStore(StringIO(SHORT_FILE_CONTENTS))
c1 = Customer('A', [Item('bananas', 7)])
c2 = Customer('B', [Item('apple', 2)])
c3 = Customer('C', [Item('orange', 5)])
c4 = Customer('D', [Item('grapes', 10)])
gs.enter_line(c1)
gs.enter_line(c2)
gs.enter_line(c3)
gs.enter_line(c4)
assert gs.line_is_ready(0) == False
assert gs.line_is_ready(1) == True
assert gs.line_is_ready(2) == True
assert gs2.line_is_ready(0) == False
assert gs2.line_is_ready(1) == False
assert gs2.line_is_ready(2) == False
def test_grocery_store_init() -> None:
""" Test store initialization with correct lines."""
store = GroceryStore(StringIO(GROCERY_STORE))
assert store._regular_count == 3
assert store._express_count == 2
assert store._self_serve_count == 4
assert len(store._lines) == 9
for line in store._lines:
assert line.capacity == 10
def test_enter_line_lines_full() -> None:
""" Test whether customer enters line with full line(s)."""
store = GroceryStore(GROCERY_STORE_1_LINE_1_CAP)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
item_list_2 = [Item('apples', 5)]
arnold = Customer('Arnold', item_list_2)
store.enter_line(arnold)
assert store.enter_line(belinda) == -1
def test_enter_first_occupied() -> None:
""" Test whether customer enters correct line when first is occupied."""
store = GroceryStore(GROCERY_STORE)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
item_list_2 = [Item('apples', 5)]
arnold = Customer('Arnold', item_list_2)
store.enter_line(arnold)
assert store.enter_line(belinda) == 1
def test_complete_checkout_more() -> None:
""" Test whether customers left to checkout after checkout when
line is empty."""
store = GroceryStore(GROCERY_STORE_1_LINE_2_CAP)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
charlie = Customer('Charlie', item_list)
store.enter_line(belinda)
store.enter_line(charlie)
assert store.complete_checkout(0)
def test_is_ready_more_occupants() -> None:
""" Test line is ready with more than one customer."""
store = GroceryStore(GROCERY_STORE_1_LINE_2_CAP)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
item_list_2 = [Item('apples', 5)]
arnold = Customer('Arnold', item_list_2)
store.enter_line(belinda)
store.enter_line(arnold)
assert not store.line_is_ready(0)
def test_enter_over_express_limit() -> None:
""" Test whether customer enters express line when they have too
many items."""
store = GroceryStore(GROCERY_STORE_EXPRESS)
item_list = [Item('apples', 5)]
for i in range(EXPRESS_LIMIT):
item_list.append(Item('apples', i))
arnold = Customer('Arnold', item_list)
store.enter_line(arnold)
assert store.enter_line(arnold) == -1
def test_cannot_enter_line() -> None:
"""test for when the line capacity is full"""
store1 = GroceryStore(StringIO(CONFIG_FILE2))
customer1 = Customer("bruce", [])
customer2 = Customer("tom", [])
customer3 = Customer("mary", [])
assert store1.enter_line(customer1) == 0
assert store1.enter_line(customer2) == 1
assert store1.enter_line(customer3) == -1
assert customer3.arrival_time == -1
def test_close_line_people_in_line() -> None:
"""check close line with people in line"""
store1 = GroceryStore(StringIO(CONFIG_FILE4))
customer1 = Customer("bruce", [])
customer2 = Customer("tom", [])
customer3 = Customer("mary", [])
assert store1.enter_line(customer1) == 0
assert store1.enter_line(customer2) == 1
assert store1.enter_line(customer3) == 0
assert store1.close_line(0) == [customer3]
assert store1.close_line(1) == []
def test_get_first_in_line() -> None:
"""check get first method"""
store1 = GroceryStore(StringIO(CONFIG_FILE4))
customer1 = Customer("bruce", [])
customer2 = Customer("tom", [])
customer3 = Customer("mary", [])
assert store1.enter_line(customer1) == 0
assert store1.enter_line(customer2) == 1
assert store1.enter_line(customer3) == 0
assert store1.get_first_in_line(0) == customer1
assert store1.get_first_in_line(1) == customer2
def test_get_first_in_line_one() -> None:
""" Test whether get first in line returns correct customer with
more than one customer in line."""
store = GroceryStore(GROCERY_STORE_1_LINE_2_CAP)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
store.enter_line(belinda)
item_list_2 = [Item('cherries', 5)]
charlie = Customer('Charlie', item_list_2)
store.enter_line(charlie)
assert store.get_first_in_line(0) == belinda
def test_close_line_more_occupants() -> None:
""" Test whether line with more than one customer returns correct
customers on close."""
store = GroceryStore(GROCERY_STORE_1_LINE_2_CAP)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
store.enter_line(belinda)
item_list_2 = [Item('cherries', 5)]
charlie = Customer('Charlie', item_list_2)
store.enter_line(charlie)
assert store.close_line(0) == [charlie]
def test_get_first_in_line():
f = StringIO(CONFIG_FILE_100_03)
store4 = GroceryStore(f)
customer1 = Customer('b', [Item('a', 29)])
customer2 = Customer('c', [Item('a', 2)])
store4.enter_line(customer1)
store4.enter_line(customer2)
assert store4.get_first_in_line(0) == customer1
assert store4.complete_checkout(0)
assert store4.get_first_in_line(0) == customer2
assert not store4.complete_checkout(0)
assert store4.get_first_in_line(0) is None
def test_start_checkout_normal_line() -> None:
"""check the checkout for normal line"""
store1 = GroceryStore(StringIO(CONFIG_FILE2))
item1 = Item("banana", 10)
item2 = Item("apple", 2)
item3 = Item("orange", 9)
customer1 = Customer("bruce", [item1, item2, item3])
customer2 = Customer("tom", [item1, item2, item3])
assert store1.enter_line(customer1) == 0
assert store1.start_checkout(0) == 21
assert store1.enter_line(customer2) == 1
assert store1.start_checkout(1) == 21
def test_enter_line_lines_occupied() -> None:
""" Test whether customer enters a line with another customer in it."""
store = GroceryStore(GROCERY_STORE_2_LINES_2_CAP)
item_list = [Item('bananas', 6)]
belinda = Customer('Belinda', item_list)
item_list_2 = [Item('apples', 5)]
arnold = Customer('Arnold', item_list_2)
item_list_3 = [Item('carrots', 3)]
charlie = Customer('Charlie', item_list_3)
store.enter_line(arnold)
store.enter_line(belinda)
assert store.enter_line(charlie) == 0
assert len(store._lines[0]) == 2
def test_enter_line2() -> None:
"""check if people can enter line correctly."""
store = GroceryStore(StringIO(CONFIG_FILE2))
customer1 = Customer('bruce', [])
customer2 = Customer('tom', [Item('banana', 1), Item('banana', 1), Item('banana', 1),
Item('banana', 1), Item('banana', 1), Item('banana', 1),
Item('banana', 1), Item('banana', 1)])
customer3 = Customer('jerry', [])
customer4 = Customer('mary', [])
assert store.enter_line(customer1) == 0
assert store.enter_line(customer2) == -1
assert store.enter_line(customer3) == 1
assert store.enter_line(customer4) == -1
def test_line_is_ready_and_close():
f = StringIO(CONFIG_FILE_100_03)
store4 = GroceryStore(f)
customer1 = Customer('b', [Item('a', 29)])
customer2 = Customer('c', [Item('a', 2)])
customer3 = Customer('d', [Item('a', 2)])
store4.enter_line(customer1)
assert store4.line_is_ready(0)
store4.enter_line(customer2)
store4.enter_line(customer3)
assert len(store4.close_line(0)) == 2
assert store4.line_is_ready(0)
assert store4.start_checkout(0) == 29
assert not store4.complete_checkout(0)
assert not store4.line_is_ready(0)
def test_complete_checkout_with_customer_behind() -> None:
"""check complete with customers behind"""
store1 = GroceryStore(StringIO(CONFIG_FILE4))
customer1 = Customer("bruce", [])
customer2 = Customer("tom", [])
customer3 = Customer("mary", [])
customer4 = Customer("jams", [])
customer5 = Customer("jerry", [])
assert store1.enter_line(customer1) == 0
assert store1.enter_line(customer2) == 1
assert store1.enter_line(customer3) == 0
assert store1.enter_line(customer4) == 1
assert store1.enter_line(customer5) == 0
assert store1.complete_checkout(0) == 2
assert store1.complete_checkout(1) == 1