def test_type_sum(): storage = { 21: [1000, "in", "salary"], 22: [[100, "out", "pizza"]], 23: [[100, "out", "pizza"], [150, "out", "lunch"], [1000, "in", "salary"]] } assert type_sum(storage, 23, "in") == 1000 assert type_sum(storage, 23, "out") == 250
def ui_list_balance(storage, day): in_tr = type_sum(storage, day, "in") out_tr = type_sum(storage, day, "out") print(in_tr - out_tr)