From 2b9aa00bd5db47ce5b34e170748f6c6c266feb97 Mon Sep 17 00:00:00 2001 From: Francis Deslauriers Date: Thu, 5 Sep 2019 18:10:54 -0400 Subject: [PATCH] bt2: field.py: add `_count` method to `_VariantField` Also, add corresponding test case. Signed-off-by: Francis Deslauriers Change-Id: I536526aa0fa1c196056f35d6a74464b076e4b9d4 Reviewed-on: https://review.lttng.org/c/babeltrace/+/2005 Tested-by: jenkins Reviewed-by: Simon Marchi --- src/bindings/python/bt2/bt2/field.py | 3 +++ tests/bindings/python/bt2/test_field.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/src/bindings/python/bt2/bt2/field.py b/src/bindings/python/bt2/bt2/field.py index 4d31f8fa..7d5fdd83 100644 --- a/src/bindings/python/bt2/bt2/field.py +++ b/src/bindings/python/bt2/bt2/field.py @@ -559,6 +559,9 @@ class _OptionField(_Field): class _VariantField(_ContainerField, _Field): _NAME = 'Variant' + def _count(self): + return len(self.cls) + @property def selected_option_index(self): return native_bt.field_variant_get_selected_option_field_index(self._ptr) diff --git a/tests/bindings/python/bt2/test_field.py b/tests/bindings/python/bt2/test_field.py index 5c8d8287..f478515f 100644 --- a/tests/bindings/python/bt2/test_field.py +++ b/tests/bindings/python/bt2/test_field.py @@ -1988,6 +1988,9 @@ class VariantFieldTestCase(unittest.TestCase): self._def.value = 1774 self.assertEqual(self._def, field) + def test_len(self): + self.assertEqual(len(self._def), 4) + def test_eq_invalid_type(self): self._def.selected_option_index = 1 self._def.value = 'gerry' -- 2.34.1