aboutsummaryrefslogtreecommitdiff
path: root/absl/flags/tests
diff options
context:
space:
mode:
authorAbseil Team <absl-team@google.com>2020-04-07 10:28:43 -0700
committerCopybara-Service <copybara-worker@google.com>2020-04-07 10:29:06 -0700
commita97bd99f120275f732b547297dee0f1bd694afe7 (patch)
tree8e964da17729724664f8831944d1cff5b04e366b /absl/flags/tests
parent2c2a0c3e7fb29fc8bebff2a1753c6c4961f65d0e (diff)
downloadabsl-py-a97bd99f120275f732b547297dee0f1bd694afe7.tar.gz
Add support for multi-flags to flag_dict_to_args helper.
Multi-flags are identified by an optional second argument consisting of a set of multi-flag names. PiperOrigin-RevId: 305287708 Change-Id: I26f1f1f40771d7e3a2d651812e6b3bf6fe3c5c17
Diffstat (limited to 'absl/flags/tests')
-rw-r--r--absl/flags/tests/flags_test.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/absl/flags/tests/flags_test.py b/absl/flags/tests/flags_test.py
index 7bb7dca..0a54cd3 100644
--- a/absl/flags/tests/flags_test.py
+++ b/absl/flags/tests/flags_test.py
@@ -67,6 +67,17 @@ class FlagDictToArgsTest(absltest.TestCase):
'--loadthatstuff=42,hello,goodbye'),
flags.flag_dict_to_args(arg_dict))
+ def test_flatten_google_flag_map_with_multi_flag(self):
+ arg_dict = {
+ 'some_list': ['value1', 'value2'],
+ 'some_multi_string': ['value3', 'value4'],
+ }
+ self.assertSameElements(
+ (
+ '--some_list=value1,value2', '--some_multi_string=value3',
+ '--some_multi_string=value4'),
+ flags.flag_dict_to_args(arg_dict, multi_flags={'some_multi_string'}))
+
class Fruit(enum.Enum):
apple = 1