Update test_worker.py

Author: justinmerrell

tests/test_serverless/test_worker.py

@@ -468,7 +468,8 @@ def mock_is_alive():
             scale_behavior['counter'] += 1
             return res
 
-        with patch("runpod.serverless.modules.rp_scale.JobScaler.is_alive", wraps=mock_is_alive):
+        with patch("runpod.serverless.modules.rp_scale.JobScaler.is_alive", wraps=mock_is_alive), \
+                patch("runpod.serverless.os", return_value=None):
             runpod.serverless.start(config)
 
     # Test with sls-core
@@ -478,7 +479,8 @@ def test_run_worker_with_sls_core(self):
         '''
         os.environ["RUNPOD_USE_CORE"] = "true"
 
-        with patch("runpod.serverless.core.main") as mock_main:
+        with patch("runpod.serverless.core.main") as mock_main, \
+                patch("runpod.serverless.os", return_value=None):
             runpod.serverless.start(self.config)
 
             assert mock_main.called