def test_floor_ceil(bank): t = symbol('t', discover(bank)) assert set(compute(200 * floor(t.amount / 200), bank)) == set([0, 200]) assert set(compute(200 * ceil(t.amount / 200), bank)) == set([200, 400])