Introduce Pytest

test/test.py

@@ -0,0 +1,48 @@
+import os
+
+import pandas as pd
+import pytest
+
+from queries.process_all_db import (
+    all_dbs,
+    process_all_tech_db,
+    process_all_tech_db_with_stats,
+)
+from utils.utils_vars import UtilsVars
+
+
+class TestProcessAllDB:
+    def setup_method(self):
+        UtilsVars.all_db_dfs = []
+        UtilsVars.final_all_database = None
+
+    def test_all_dbs(self):
+        filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\20240917_7950_17092024_Dump.xml.gz.xlsb"
+        all_dbs(filepath)
+        assert len(UtilsVars.all_db_dfs) == 7
+        assert isinstance(UtilsVars.all_db_dfs[0], pd.DataFrame)
+
+    def test_process_all_tech_db(self):
+        filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\20240917_7950_17092024_Dump.xml.gz.xlsb"
+        process_all_tech_db(filepath)
+        assert UtilsVars.final_all_database is not None
+
+    def test_process_all_tech_db_with_stats(self):
+        filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\20240917_7950_17092024_Dump.xml.gz.xlsb"
+        process_all_tech_db_with_stats(filepath)
+        assert UtilsVars.final_all_database is not None
+
+    # def test_all_dbs_empty_file(self):
+    #     filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\empty.xlsb"
+    #     all_dbs(filepath)
+    #     assert len(UtilsVars.all_db_dfs) == 0
+
+    # def test_process_all_tech_db_empty_file(self):
+    #     filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\empty.xlsb"
+    #     process_all_tech_db(filepath)
+    #     assert UtilsVars.final_all_database is None
+
+    # def test_process_all_tech_db_with_stats_empty_file(self):
+    #     filepath = r"C:\Users\HP\Desktop\LTE\PROJET 2023\DUMP\2024\SEPTEMBRE\empty.xlsb"
+    #     process_all_tech_db_with_stats(filepath)
+    #     assert UtilsVars.final_all_database is None

utils/utils_vars.py

@@ -44,7 +44,7 @@ class UtilsVars:
     final_mrbts_database = ""
     final_mal_database = ""
     all_db_dfs = []
-    final_all_database = ""
+    final_all_database = None
     neighbors_database = ""
     file_path = ""
     physisal_db = get_physical_db()