def test_variant_generator(self):
vg = VariantGenerator()
vg.add('key1', [1, 2, 3])
vg.add('key2', [True, False])
vg.add('key3', lambda key2: [1] if key2 else [1, 2])
assert len(vg.variants()) == 9
class VG(VariantGenerator):
@variant
def key1(self):
return [1, 2, 3]
@variant
def key2(self):
yield True
yield False
@variant
def key3(self, key2):
if key2:
yield 1
else:
yield 1
yield 2
assert len(VG().variants()) == 9
def test_variant_generator():
vg = VariantGenerator()
vg.add("key1", [1, 2, 3])
vg.add("key2", [True, False])
vg.add("key3", lambda key2: [1] if key2 else [1, 2])
it.assertEqual(len(vg.variants()), 9)
class VG(VariantGenerator):
@variant
def key1(self):
return [1, 2, 3]
@variant
def key2(self):
yield True
yield False
@variant
def key3(self, key2):
if key2:
yield 1
else:
yield 1
yield 2
it.assertEqual(len(VG().variants()), 9)