def test_contract_tuple(): assert contract_tuple((1, 1, 2, 3, 1), 2) == (2, 2, 2, 2) assert contract_tuple((1, 1, 2, 5, 1), 2) == (2, 2, 4, 2) assert contract_tuple((2, 4), 2) == (2, 4) assert contract_tuple((2, 4), 3) == (6,)