Fix expected key name

gyorilab · Oct 30, 2023 · a7596a3 · a7596a3
1 parent 255494c
commit a7596a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_ops.py b/tests/test_ops.py
@@ -219,7 +219,7 @@ def test_stratify_w_name_map(self):
         self.assertIn(original_name, sir_parameterized.initials)
         for city in cities:
             city_name = city_name_map.get(city, city)
-            key = f"{original_name}_{city_name}".replace(':', '_').replace(' ', '_')
+            key = f"{original_name}_{city_name}".replace(':', '_')
             self.assertIn(key, actual.initials, msg=f"Key '{key}' not in initials")
             # Cannot use .args[0] here as .args[0] not a primitive data type
             self.assertEqual(