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