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