From a4465b73472f6fab294a30c6a07ab4b6b9b3e908 Mon Sep 17 00:00:00 2001 From: eccoope <125493409+eccoope@users.noreply.github.com> Date: Mon, 15 Apr 2024 14:42:57 -0600 Subject: [PATCH] Added argument to test_categorize() to reflect changes made to categorize() --- pvanalytics/tests/features/test_snow.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pvanalytics/tests/features/test_snow.py b/pvanalytics/tests/features/test_snow.py index 6e8ca834..376479d1 100644 --- a/pvanalytics/tests/features/test_snow.py +++ b/pvanalytics/tests/features/test_snow.py @@ -67,7 +67,8 @@ def test_get_transmission(): def test_categorize(): vmp_ratio = np.array([np.nan, 0.9, 0.1, 0.6, 0.7, 0.9, 0.9]) - voltage = np.array([400., 400., 400., 400., 400., 400., 200.]) + measured_voltage = np.array([400., 450., 400., 420., 420., 495., 270.]) + modeled_voltage = np.array([np.nan, 500, 4000, 700, 600, 550, 300]) transmission = np.array([0.5, np.nan, 0.9, 0.9, 0.5, 0.9, 0.9]) min_dcv = 300 threshold_vratio = 0.7 @@ -76,6 +77,7 @@ def test_categorize(): # vo>thres, vo>thres, vo>thres, vo>thres, vo>thres, vo>thres, vothres, trthres expected = np.array([None, None, 2, 2, 3, 4, 0]) - result = snow.categorize(vmp_ratio, transmission, voltage, min_dcv, + result = snow.categorize(vmp_ratio, transmission, measured_voltage, + modeled_voltage, min_dcv, threshold_vratio, threshold_transmission) assert_array_equal(result, expected)