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_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])
