From e5903842ba4853fa66e82f08aa34de05ac8570d1 Mon Sep 17 00:00:00 2001
From: Ananta Shrestha <anshrest@asu.edu>
Date: Tue, 12 Nov 2024 09:52:02 -0700
Subject: [PATCH] Initial test cases for scaffolding.py

- Executes all the unit test files available in the tests directory
- Test get_time_query() with the use of patch for LocalData and TimeComponentQuery
- Basic test case to assert cases for MODE mapping to en-translations
---
 viz_scripts/docker/start_tests.sh     |  2 +-
 viz_scripts/tests/test_scaffolding.py | 66 +++++++++++++++++++++++++++
 2 files changed, 67 insertions(+), 1 deletion(-)
 create mode 100644 viz_scripts/tests/test_scaffolding.py

diff --git a/viz_scripts/docker/start_tests.sh b/viz_scripts/docker/start_tests.sh
index f979f23..2ae0c46 100644
--- a/viz_scripts/docker/start_tests.sh
+++ b/viz_scripts/docker/start_tests.sh
@@ -8,6 +8,6 @@ conda env list
 cd saved-notebooks/tests || exit 1
 
 echo "Starting unit tests..."
-PYTHONPATH=../.. coverage run -m pytest test_plots.py -v
+PYTHONPATH=../.. coverage run -m pytest . -v
 
 coverage report
diff --git a/viz_scripts/tests/test_scaffolding.py b/viz_scripts/tests/test_scaffolding.py
new file mode 100644
index 0000000..2af1f15
--- /dev/null
+++ b/viz_scripts/tests/test_scaffolding.py
@@ -0,0 +1,66 @@
+import unittest.mock as mock
+import emission.core.wrapper.localdate as ecwl
+import emission.storage.timeseries.tcquery as esttc
+import importlib
+import pandas as pd
+import numpy as np
+import collections as colls
+
+# Dynamically import saved-notebooks.plots
+scaffolding = importlib.import_module('saved-notebooks.scaffolding')
+
+@mock.patch('emission.core.wrapper.localdate.LocalDate')
+@mock.patch('emission.storage.timeseries.tcquery.TimeComponentQuery')
+def test_get_time_query(mock_tcquery, mock_localdate):
+    mock_localdate.return_value = ecwl.LocalDate({"year": 2022, "month": 6})
+    mock_tcquery.return_value = esttc.TimeComponentQuery("data.start_local_dt", mock_localdate.return_value, mock_localdate.return_value)
+
+    result = scaffolding.get_time_query(2022, 6)
+
+    assert result == mock_tcquery.return_value
+    mock_localdate.assert_called_with({"year": 2022, "month": 6})
+    mock_tcquery.assert_called_with("data.start_local_dt", mock_localdate.return_value, mock_localdate.return_value)
+
+def test_mapping_labels():
+    dynamic_labels = {
+        "MODE": [
+            {"value":"gas_car", "base_mode": "CAR",
+            "baseMode":"CAR", "met_equivalent":"IN_VEHICLE", "kgCo2PerKm": 0.22031},
+            {"value":"motorcycle", "base_mode": "MOPED", "footprint": { "gasoline": { "wh_per_km": 473.17 }},
+            "baseMode":"MOPED", "met_equivalent":"IN_VEHICLE", "kgCo2PerKm": 0.113143309},
+            {"value":"walk", "base_mode": "WALKING",
+            "baseMode":"WALKING", "met_equivalent":"WALKING", "kgCo2PerKm": 0},
+            {"value":"e_car", "base_mode": "E_CAR",
+            "baseMode":"E_CAR", "met_equivalent":"IN_VEHICLE", "kgCo2PerKm": 0.08216},
+            {"value":"taxi", "base_mode": "TAXI",
+            "baseMode":"TAXI", "met_equivalent":"IN_VEHICLE", "kgCo2PerKm": 0.30741},
+            {"value":"bike", "base_mode": "BICYCLING",
+            "baseMode":"BICYCLING", "met_equivalent":"BICYCLING", "kgCo2PerKm": 0},
+            {"value":"air", "base_mode": "AIR",
+            "baseMode":"AIR", "met_equivalent":"IN_VEHICLE", "kgCo2PerKm": 0.09975}
+        ],
+        "translations": {
+            "en": {
+            "walk": "Walk",
+            "motorcycle":"Motorcycle",
+            "bike": "Bicycle",
+            "gas_car": "Car",
+            "e_car": "Electric Car",
+            "taxi": "Taxi",
+            "air": "Airplane"
+            }
+        }
+    }
+
+    result_mode = scaffolding.mapping_labels(dynamic_labels, "MODE")
+
+    expected_result_mode = colls.defaultdict(lambda: 'Other', {
+        "gas_car": "Car",
+        "motorcycle": "Motorcycle",
+        "walk": "Walk",
+        "e_car": "Electric Car",
+        "taxi": "Taxi",
+        "bike": "Bicycle",
+        "air": "Airplane"
+    })
+    assert result_mode == expected_result_mode
\ No newline at end of file