Update UnitTesting for figure generation

UnitTests/test_fittingfunc.py

@@ -1,4 +1,5 @@
 import os
+import glob
 import unittest
 import numpy as np
 import pandas as pd
@@ -201,8 +202,12 @@ def test_MCMC_exportpath(self):
                     temp_export_path,
                     "output.csv")))
 
+        figure_files = glob.glob(os.path.join(temp_export_path, "*.pdf"))
+        self.assertTrue(len(figure_files) > 0, "No figure files found in the export path.")
+
         # Cleanup
-        os.remove(os.path.join(temp_export_path, "output.csv"))
+        for file_path in glob.glob(os.path.join(temp_export_path, "*")):
+            os.remove(file_path)
         os.rmdir(temp_export_path)
 
 

UnitTests/test_loading.py

@@ -86,5 +86,10 @@ def test_chemistry_thickness_path_none(self):
             loader = pig.SampleDataLoader(chemistry_thickness_path=None)
             loader.load_chemistry_thickness()
 
+    def test_spectrum_chemistry_thickness_path_none(self):
+        with self.assertRaises(ValueError):
+            loader = pig.SampleDataLoader(spectrum_path=None, chemistry_thickness_path=None)
+            loader.load_all_data()
+
 if __name__ == '__main__':
     unittest.main()

UnitTests/test_thickness.py

@@ -123,10 +123,5 @@ def test_process_thickness(self):
         except Exception as e:
             self.fail(f"Unexpected exception occurred: {e}")
 
-    def test_invalid_wavenumber_range(self):
-
-        with self.assertRaises(ValueError):
-            pig.calculate_mean_thickness(self.dfs_dict, 1.546, self.wn_high_gl, self.wn_low_gl)
-
 if __name__ == '__main__':
     unittest.main()