change quicklooks test, no flight_id for quicklooks for now

Geet-George · Aug 10, 2024 · f2a69c3 · f2a69c3
1 parent 0387f93
commit f2a69c3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/halodrops/helper/paths.py b/src/halodrops/helper/paths.py
@@ -82,6 +82,7 @@ def __init__(
         """
 
         self.path_structure = path_structure
+        self.data_directory = data_directory
 
         if self.path_structure == "flightid_first":
             self.logger = logging.getLogger("halodrops.helper.paths.Paths")
@@ -135,7 +136,9 @@ def quicklooks_path(self):
         `str`
             Path to quicklooks directory
         """
-        quicklooks_path_str = os.path.join(self.flight_idpath, "Quicklooks")
+        if self.path_structure == "levels_first":
+            quicklooks_path_str = os.path.join(self.data_directory, self.platform_id, "Quicklooks")
+
         if pp(quicklooks_path_str).exists():
             self.logger.info(f"Path exists: {quicklooks_path_str=}")
         else:

diff --git a/tests/test_paths.py b/tests/test_paths.py
@@ -8,11 +8,13 @@
 
 if path_structure == "levels_first":
     l1_path = os.path.join(main_data_directory, platform, "Level_1", flightdate)
+    quicklooks_path = os.path.join(main_data_directory, platform, "Quicklooks")
+
 if path_structure == "flightid_first":
     l1_path = os.path.join(main_data_directory, platform, flightdate, "Level_1")
-
-quicklooks_path = os.path.join(main_data_directory, platform, flightdate, "Quicklooks")
+    quicklooks_path = os.path.join(main_data_directory, platform, flightdate, "Quicklooks")
 
+
 object = paths.Flight(main_data_directory, flightdate, platform)