Add test for indexing

OpenMined · Oct 19, 2023 · 8bab166 · 8bab166
1 parent 57a49c9
commit 8bab166
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/syft/tests/syft/types/dicttuple_test.py b/packages/syft/tests/syft/types/dicttuple_test.py
@@ -21,6 +21,14 @@ def test_should_iter_over_value(dict_tuple: DictTuple) -> None:
     assert values == [1, 2]
 
 
+@pytest.mark.parametrize("dict_tuple", TEST_CASES)
+def test_indexing(dict_tuple: DictTuple) -> None:
+    assert dict_tuple[0] == 1
+    assert dict_tuple[1] == 2
+    assert dict_tuple["x"] == 1
+    assert dict_tuple["y"] == 2
+
+
 @pytest.mark.parametrize("dict_tuple", TEST_CASES)
 def test_convert_to_other_iterable_types(dict_tuple: DictTuple) -> None:
     assert list(dict_tuple) == [1, 2]