fix: exclude build directory in tests

Author: justinmerrell

.github/workflows/CI-pylint.yml

@@ -34,4 +34,4 @@ jobs:
 
       - name: Pylint Source
         run: |
-          find . -type f -name '*.py' | xargs pylint --extension-pkg-whitelist='pydantic'
+          find . -type f -name '*.py' | grep -v './build/' | xargs pylint --extension-pkg-whitelist='pydantic'

.github/workflows/CI-tests.yml

@@ -34,4 +34,4 @@ jobs:
 
       - name: Run Tests
         run: |
-          pytest --timeout=120 --timeout_method=thread --cov=runpod --cov-report=xml --cov-report=term-missing --cov-fail-under=100 -W error -p no:cacheprovider -p no:unraisableexception
+          pytest --ignore=build --timeout=120 --timeout_method=thread --cov=runpod --cov-report=xml --cov-report=term-missing --cov-fail-under=100 -W error -p no:cacheprovider -p no:unraisableexception

tests/test_version.py

@@ -6,12 +6,14 @@
 
 def test_version_found():
     """ Test that the version is found """
-    with patch('runpod.version.get_distribution', return_value=Mock(version='1.0.0')) as mock_get_distribution:
+    with patch('runpod.version.get_distribution') as mock_get_distribution:
+        mock_get_distribution.return_value = Mock(version='1.0.0')
         assert get_version() == '1.0.0'
         assert mock_get_distribution.called
 
 def test_version_not_found():
     """ Test that the version is not found """
-    with patch('runpod.version.get_distribution', side_effect=DistributionNotFound) as mock_get_distribution:
+    with patch('runpod.version.get_distribution') as mock_get_distribution:
+        mock_get_distribution.side_effect = DistributionNotFound
         assert get_version() == 'unknown'
         assert mock_get_distribution.called