fix: missing coverage

Author: justinmerrell

CHANGELOG.md

@@ -10,6 +10,13 @@
 
 - Reduced polling when checking for job completion.
 - Removed print statements for endpoint calls.
+- Serverless progress updates no longer restricted to only strings.
+
+## Changed
+
+- Removed `pillow` dependency.
+- Removed `python-dotenv` dependency.
+- Removed `setuptools_scm` from required dependencies.
 
 ---
 

tests/test_endpoint/test_runner.py

@@ -3,7 +3,6 @@
 '''
 
 import unittest
-from unittest import mock
 from unittest.mock import patch, Mock
 import requests
 
@@ -194,6 +193,23 @@ def test_run_sync(self, mock_client_request):
 
         self.assertEqual(run_request, {"result": "YOUR_MODEL_OUTPUT_VALUE"})
 
+    @patch.object(runpod.endpoint.runner.RunPodClient, '_request')
+    def test_run_sync_with_timeout(self, mock_client_request):
+        '''
+        Tests Endpoint.run_sync with timeout
+        '''
+        mock_client_request.return_value = {
+            "id": "123",
+            "status": "IN_PROGRESS"
+        }
+
+        runpod.api_key = "MOCK_API_KEY"
+        endpoint = runpod.Endpoint("ENDPOINT_ID")
+
+        request_data = {"YOUR_MODEL_INPUT_JSON": "YOUR_MODEL_INPUT_VALUE"}
+        with self.assertRaises(TimeoutError):
+            endpoint.run_sync(request_data, timeout=1)
+
 
 class TestJob(unittest.TestCase):
     ''' Tests for Job '''
@@ -245,6 +261,19 @@ def test_output_with_sleep(self, mock_client):
 
         self.assertEqual(output, "Job output")
 
+    @patch('runpod.endpoint.runner.RunPodClient')
+    def test_output_timeout(self, mock_client):
+        '''
+        Tests Job.output with timeout
+        '''
+        mock_client.get.return_value = {
+            "status": "IN_PROGRESS"
+        }
+
+        job = runner.Job("endpoint_id", "job_id", mock_client)
+        with self.assertRaises(TimeoutError):
+            job.output(timeout=1)
+
     @patch('runpod.endpoint.runner.RunPodClient')
     def test_job_status(self, mock_client):
         '''
@@ -262,3 +291,4 @@ def test_job_status(self, mock_client):
         job = runner.Job("endpoint_id", "job_id", mock_client)
         self.assertEqual(job.status(), "IN_PROGRESS")
         self.assertEqual(job.status(), "COMPLETED")
+        self.assertEqual(job.status(), "COMPLETED")