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)
def get_variants(args):
env_params = ENV_PARAMS[args.env]
params = SHARED_PARAMS
params.update(env_params)
vg = VariantGenerator()
for key, val in params.items():
if isinstance(val, list):
vg.add(key, val)
else:
vg.add(key, [val])
return vg
def get_variants(args):
env_params = ENV_PARAMS[args.env]
params = SHARED_PARAMS
params.update(env_params)
vg = VariantGenerator()
for key, val in params.items():
if isinstance(val, list):
vg.add(key, val)
else:
vg.add(key, [val])
vg.add('snapshot1', (args.snapshot1, ))
vg.add('snapshot2', (args.snapshot2, ))
return vg