Fixed test failure in Python 3.7

zappa · Dec 27, 2022 · 09b7a0e · 09b7a0e
1 parent 8f00757
commit 09b7a0e
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/tests/tests.py b/tests/tests.py
@@ -1113,11 +1113,19 @@ def test_load_extended_settings(self):
         self.assertEqual(True, zappa_cli.stage_config["touch"])
         self.assertIn("x86_64", zappa_cli.architecture)
 
-        zappa_cli = ZappaCLI()
-        zappa_cli.api_stage = "arch_arm64"
-        zappa_cli.load_settings("test_settings.json")
-        self.assertIn("arm64", zappa_cli.stage_config["architecture"])
-        self.assertIn("arm64", zappa_cli.architecture)
+        if sys.version_info.major == 3 and sys.version_info.minor < 8:
+            with self.assertRaises(ValueError):
+                zappa_cli = ZappaCLI()
+                zappa_cli.api_stage = "arch_arm64"
+                zappa_cli.load_settings("test_settings.json")
+                self.assertIn("arm64", zappa_cli.stage_config["architecture"])
+                self.assertIn("arm64", zappa_cli.architecture)
+        else:
+            zappa_cli = ZappaCLI()
+            zappa_cli.api_stage = "arch_arm64"
+            zappa_cli.load_settings("test_settings.json")
+            self.assertIn("arm64", zappa_cli.stage_config["architecture"])
+            self.assertIn("arm64", zappa_cli.architecture)
 
         zappa_cli = ZappaCLI()
         zappa_cli.api_stage = "extendofail"