diff --git a/providers/base/tests/test_camera_quality.py b/providers/base/tests/test_camera_quality.py index 5d3198493..fc18c2b05 100644 --- a/providers/base/tests/test_camera_quality.py +++ b/providers/base/tests/test_camera_quality.py @@ -27,7 +27,9 @@ from bin.camera_quality_test import brisque -data_dir = os.environ['PLAINBOX_PROVIDER_DATA'] +default_dir = os.path.join(os.path.dirname(__file__), '../data') +data_dir = os.getenv('PLAINBOX_PROVIDER_DATA', default=default_dir) + class CameraQualityTests(unittest.TestCase): """This class provides test cases for the camera_quality_test module.""" @@ -83,5 +85,4 @@ def test_invalid_image(self, mock_capture): mock_capture.return_value.grab.return_value = True mock_capture.return_value.read.return_value = (True, img) - assert brisque() == 1