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 test_frequency(self):
k = Gen.frequency([
(1, default.Bool().generator),
(2, default.Bool().generator)
])
ks = Gen.sample(1, 10, k)
self.assertTrue(True in ks)
self.assertTrue(False in ks)
def test_list_of(self):
s = Gen.list_of(Gen.constant(1))
ss = Gen.sample(1, 10, s)
lens = set()
for v in ss:
lens.add(len(v))
for i in v:
self.assertEqual(1, i)
self.assertTrue(0 < len(lens) < 4)
for l in lens:
self.assertTrue(l in {0, 1, 2})
def test_int(self):
size = 100
x = default.Int()
v = -1
# test generator
rs = Gen.resize(size, x.generator)
s = Gen.sample(1, size, rs)
for i in s:
self.assertTrue(-size <= i <= size)
# test shrinker
sh = x.shrinker(v)
for i in sh:
self.assertTrue(i <= abs(i))
def test_bool(self):
x = default.Bool()
s = Gen.sample(1, 10, x.generator)
self.assertTrue(True in s)
self.assertTrue(False in s)
def test_list_of_length(self):
s = Gen.list_of_length(2, Gen.constant(1))
ss = Gen.sample(1, 10, s)
for v in ss:
self.assertEqual(1, v[0])
self.assertEqual(1, v[1])
def test_sequence(self):
s = Gen.sequence([Gen.constant(1), Gen.constant(3)])
ss = Gen.sample(1, 10, s)
for v in ss:
self.assertEqual(1, v[0])
self.assertEqual(3, v[1])
def test_oneof(self):
k = Gen.oneof([default.Bool().generator, default.Bool().generator])
ks = Gen.sample(1, 10, k)
self.assertTrue(True in ks)
self.assertTrue(False in ks)
