From 0fd4dd0932b01f54fee061b8d385949a7c3a43c2 Mon Sep 17 00:00:00 2001
From: jpfleischer <jacquespfleischer@gmail.com>
Date: Wed, 23 Oct 2024 13:20:26 -0400
Subject: [PATCH] Add 2021 NYC ebike test

---
 test/metrics/test_footprint_calculations.py | 32 +++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/test/metrics/test_footprint_calculations.py b/test/metrics/test_footprint_calculations.py
index e36594e..5d53bc3 100644
--- a/test/metrics/test_footprint_calculations.py
+++ b/test/metrics/test_footprint_calculations.py
@@ -125,6 +125,38 @@ async def test_nyc_bus_footprint():
         expectEqual(metadata[key], expected_metadata[key])
 
 
+@jest_test
+async def test_2021_nyc_ebike_footprint():
+    """
+    Test kWh and kg CO2e for a 2021 ebike in NYC.
+    """
+    fake_trip = {
+        "_id": "fake_trip_id",
+        "distance": 1000,
+        "start_fmt_time": "2021-01-01",
+        "start_loc": {"coordinates": [-74.006, 40.7128]},
+        "user_input": {"mode_confirm": "ebike"}
+    }
+    fake_label_options = {
+        "MODE": [
+            {"value": "ebike", "base_mode": "E_BIKE"}
+        ]
+    }
+
+    (footprint, metadata) = await emcmff.calc_footprint_for_trip(fake_trip, fake_label_options)
+
+    expected_footprint = {'kwh': 0.01367, 'kg_co2': 0.005071}
+    expected_metadata = {
+        "data_sources": ["egrid2021"],
+        "is_provisional": False,
+        "requested_year": 2021,
+    }
+    for key in expected_footprint.keys():
+        expectAlmostEqual(footprint[key], expected_footprint[key], places=2)
+    for key in expected_metadata.keys():
+        expectEqual(metadata[key], expected_metadata[key])
+
+
 @jest_test
 async def test_impact_of_ebike_replacing_car():
     """