Added default dir for the data

canonical · Oct 16, 2023 · d485d06 · d485d06
1 parent 2ba2d25
commit d485d06
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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