def test_sized(self): def test(a): return Gen(lambda n, r: a + 3) k = Gen.sized(test) ks = Gen.sample(1, 10, k) self.assertTrue(3 in ks) self.assertTrue(4 in ks)
def Int(): a = Arbitrary() a.generator = Gen.sized(lambda n: Gen.choose(-n, n)) a.shrinker = lambda n: shrink_number(n) return a