diff options
Diffstat (limited to 'absl/flags/tests/flags_test.py')
-rw-r--r-- | absl/flags/tests/flags_test.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/absl/flags/tests/flags_test.py b/absl/flags/tests/flags_test.py index 90094d3..ac77707 100644 --- a/absl/flags/tests/flags_test.py +++ b/absl/flags/tests/flags_test.py @@ -1184,6 +1184,26 @@ class MultiNumericalFlagsTest(absltest.TestCase): expected_floats, FLAGS.get_flag_value('m_float', None)): self.assertAlmostEqual(expected, actual) + def test_multi_numerical_with_tuples(self): + """Verify multi_int/float accept tuples as default values.""" + flags.DEFINE_multi_integer( + 'm_int_tuple', + (77, 88), + 'integer option that can occur multiple times', + short_name='mi_tuple') + self.assertListEqual(FLAGS.get_flag_value('m_int_tuple', None), [77, 88]) + + dict_with_float_keys = {2.2: 'hello', 3: 'happy'} + float_defaults = dict_with_float_keys.keys() + flags.DEFINE_multi_float( + 'm_float_tuple', + float_defaults, + 'float option that can occur multiple times', + short_name='mf_tuple') + for (expected, actual) in zip(float_defaults, + FLAGS.get_flag_value('m_float_tuple', None)): + self.assertAlmostEqual(expected, actual) + def test_single_value_default(self): """Test multi_int and multi_float flags with a single default value.""" int_default = 77 |