ternary

gsweene2 · Mar 12, 2021 · 8fbc1f1 · 8fbc1f1
1 parent 5533afb
commit 8fbc1f1
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 1 deletion.
diff --git a/__pycache__/python_utilities.cpython-37.pyc b/__pycache__/python_utilities.cpython-37.pyc
diff --git a/python_utilities.py b/python_utilities.py
@@ -18,4 +18,8 @@ def deep_copy(my_list):
 
     @staticmethod
     def get_key_from_map_else_return_default(my_map, key):
-        return my_map.get(key, 'Not Found')
+        return my_map.get(key, 'Not Found')
+
+    @staticmethod
+    def get_key_from_map_else_return_default_ternary(my_map, key):
+        return my_map[key] if key in my_map else 'Not Found'
diff --git a/test_python_utilities.py b/test_python_utilities.py
@@ -68,5 +68,22 @@ def test_get_key_from_map_else_return_default_key_dne(self):
         # Assert
         self.assertEqual(value, 'Not Found')
 
+
+    def test_get_key_from_map_else_return_default_ternary_key_exists(self):
+        # Arrange
+        my_map = {'a':1,'b':2,'c':3}
+        # Act
+        value = Utilities.get_key_from_map_else_return_default_ternary(my_map, 'a')
+        # Assert
+        self.assertEqual(value, 1)
+
+    def test_get_key_from_map_else_return_default_ternary_key_dne(self):
+        # Arrange
+        my_map = {'a':1,'b':2,'c':3}
+        # Act
+        value = Utilities.get_key_from_map_else_return_default_ternary(my_map, 'z')
+        # Assert
+        self.assertEqual(value, 'Not Found')
+
 if __name__ == '__main__':
     unittest.main()