예제 #1
0
 def test_no_merge_strategy(self):
   args = self._create_mock_args(
       variant_merge_strategy=MergeOptions.NONE,
       info_keys_to_move_to_calls_regex=None,
       copy_quality_to_calls=None,
       copy_filter_to_calls=None)
   self.assertEqual(_get_variant_merge_strategy(args), None)
예제 #2
0
 def test_valid_merge_strategy(self):
   args = self._create_mock_args(
       variant_merge_strategy=MergeOptions.MOVE_TO_CALLS,
       info_keys_to_move_to_calls_regex=None,
       copy_quality_to_calls=None,
       copy_filter_to_calls=None)
   self.assertIsInstance(_get_variant_merge_strategy(args),
                         move_to_calls_strategy.MoveToCallsStrategy)
예제 #3
0
 def test_invalid_merge_strategy_raises_error(self):
   args = self._create_mock_args(
       variant_merge_strategy='NotAMergeStrategy',
       info_keys_to_move_to_calls_regex=None,
       copy_quality_to_calls=None,
       copy_filter_to_calls=None)
   with self.assertRaises(ValueError):
     _ = _get_variant_merge_strategy(args)