From cfd9579171f32498fff3a2e94305699badfbcd92 Mon Sep 17 00:00:00 2001
From: samgdotson <sdotson@ucsusa.org>
Date: Wed, 13 Mar 2024 17:02:05 -0500
Subject: [PATCH] adds tests for atb class

---
 nrelpy/tests/test_atb.py | 67 +++++++++++++++++++++++-----------------
 1 file changed, 39 insertions(+), 28 deletions(-)

diff --git a/nrelpy/tests/test_atb.py b/nrelpy/tests/test_atb.py
index 6d920f8..f8ebb9e 100644
--- a/nrelpy/tests/test_atb.py
+++ b/nrelpy/tests/test_atb.py
@@ -1,6 +1,8 @@
 from nrelpy.atb import as_dataframe, ATBe
+import pandas as pd
 from urllib.error import HTTPError
 import pytest
+import numpy as np
 
 good_year = 2020
 bad_year = -999
@@ -85,39 +87,48 @@ def test_as_dataframe_bad_database():
 
 
 def test_ATB_init():
-
-    atb_class = ATBe(year=2022)
-
-    assert atb_class.scenario == 'Moderate'
-    assert atb_class.case == 'R&D'
-    assert atb_class.year == 2022
-    assert atb_class.dataframe.scenario.unique() == ['Moderate']
-
-    return
-
-def test_ATB_scenario_change():
+    atb_class = ATBe(year=good_year)
+    assert atb_class.year == good_year
     
-    atb_class = ATBe(year=2022)
+    with pytest.raises(HTTPError) as e:
+        atb_class = ATBe(bad_year)
+        
+    with pytest.raises(HTTPError) as e:
+        atb_class = ATBe(nonexistent_year)
 
-    assert atb_class.dataframe.scenario.unique() == ['Moderate']
-    assert atb_class.dataframe.core_metric_case.unique() == ['R&D']
+    return
 
-    atb_class.scenario = 'Advanced'
-    atb_class.case = 'Market'
-    assert atb_class.dataframe.scenario.unique() == ['Advanced']
-    assert atb_class.dataframe.core_metric_case.unique() == ['Market']
 
+def test_ATB_acronyms():
+    
+    atb_class = ATBe(2023)
+    assert isinstance(atb_class.acronyms, pd.DataFrame)
+    
+    with pytest.warns(RuntimeWarning) as w:
+        atb_class = ATBe(good_year)
+        acros_df = atb_class.acronyms
+    
+    return
 
 
-atb_class = ATBe(year=2022)
-def test_ATB_get_data_good_values():
-    exp_value = 7988.95108
-    actual_value = atb_class.get_data(good_tech, good_metric, good_detail)
-    assert (abs(exp_value - actual_value) < 1e-5)
+def test_ATB_access():
+    atb_class = ATBe(good_year)
+    
+    subset = atb_class(technology='Nuclear')
+    assert subset.shape == (6534,2)
+    return
+    
+def test_ATB_value_access():
+    atbe2020 = ATBe(good_year)
+    mask = (
+        (atbe2020.raw_dataframe['technology']=='Nuclear')&
+        (atbe2020.raw_dataframe['core_metric_parameter']=='LCOE')&
+        (atbe2020.raw_dataframe['core_metric_case']=='Market')&
+        (atbe2020.raw_dataframe['core_metric_variable']==2020)&
+        (atbe2020.raw_dataframe['crpyears']==20)
+        )
+    value = atbe2020.raw_dataframe[mask]['value'].values[0]
+    assert np.isclose(value, 88.22242)
+    
 
-def test_ATB_get_data_bad_values():
-    with pytest.raises(AssertionError):
-        atb_class.get_data(bad_tech, good_metric, good_detail)
-        atb_class.get_data(good_tech, bad_metric, good_detail)
-        atb_class.get_data(good_tech, good_metric, bad_detail)