def test_del_equipment(): make_test_file() ft = FTInventory("test.csv") with open('test.csv') as f: count = sum(1 for _ in f) ft.input = lambda _: ('1') ft.del_equipment() with open('test.csv') as f: count1 = sum(1 for _ in f) assert count1 == count -1
def test_checkout_equipment(): make_test_file() ft = FTInventory("test.csv") ft.input = lambda _: ('1') ft.checkout_equipment()