Skip to content

Commit

Permalink
update the routine for pytest assertions
Browse files Browse the repository at this point in the history
  • Loading branch information
kperrynrel committed Nov 6, 2023
1 parent 2b38a37 commit fc44c2d
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions pvanalytics/tests/features/test_daytime.py
Original file line number Diff line number Diff line change
Expand Up @@ -254,8 +254,8 @@ def test_get_sunrise_left_alignment(daytime_mask_left_aligned):
sunrise_left_aligned = daytime.get_sunrise(daytime_mask_left_aligned,
data_alignment='L')
# Assert that the output time series index is the same as the input
assert pd.testing.assert_index_equal(sunrise_left_aligned.index,
daytime_mask_left_aligned.index)
pd.testing.assert_index_equal(sunrise_left_aligned.index,
daytime_mask_left_aligned.index)
# Check that the output matches expected
sunrise_3_19 = sunrise_left_aligned[sunrise_left_aligned.index.date ==
date(2022, 3, 19)]
Expand All @@ -267,8 +267,8 @@ def test_get_sunrise_center_alignment(daytime_mask_center_aligned):
sunrise_center_aligned = daytime.get_sunrise(daytime_mask_center_aligned,
data_alignment='C')
# Assert that the output time series index is the same as the input
assert pd.testing.assert_index_equal(sunrise_center_aligned.index,
daytime_mask_center_aligned.index)
pd.testing.assert_index_equal(sunrise_center_aligned.index,
daytime_mask_center_aligned.index)
# Check that the output matches expected
sunrise_3_19 = sunrise_center_aligned[sunrise_center_aligned.index.date ==
date(2022, 3, 19)]
Expand All @@ -293,8 +293,8 @@ def test_get_sunset_left_alignment(daytime_mask_left_aligned):
sunset_left_aligned = daytime.get_sunset(daytime_mask_left_aligned,
data_alignment='L')
# Assert that the output time series index is the same as the input
assert pd.testing.assert_index_equal(sunset_left_aligned.index,
daytime_mask_left_aligned.index)
pd.testing.assert_index_equal(sunset_left_aligned.index,
daytime_mask_left_aligned.index)
# Check that the output matches expected
sunset_3_19 = sunset_left_aligned[sunset_left_aligned.index.date ==
date(2022, 3, 19)]
Expand All @@ -306,8 +306,8 @@ def test_get_sunset_center_alignment(daytime_mask_center_aligned):
sunset_center_aligned = daytime.get_sunset(daytime_mask_center_aligned,
data_alignment='C')
# Assert that the output time series index is the same as the input
assert pd.testing.assert_index_equal(sunset_center_aligned.index,
daytime_mask_center_aligned.index)
pd.testing.assert_index_equal(sunset_center_aligned.index,
daytime_mask_center_aligned.index)
# Check that the output matches expected
sunset_3_19 = sunset_center_aligned[sunset_center_aligned.index.date ==
date(2022, 3, 19)]
Expand All @@ -319,8 +319,8 @@ def test_get_sunset_right_alignment(daytime_mask_right_aligned):
sunset_right_aligned = daytime.get_sunset(daytime_mask_right_aligned,
data_alignment='R')
# Assert that the output time series index is the same as the input
assert pd.testing.assert_index_equal(sunset_right_aligned.index,
daytime_mask_right_aligned.index)
pd.testing.assert_index_equal(sunset_right_aligned.index,
daytime_mask_right_aligned.index)
# Check that the output matches expected
sunset_3_19 = sunset_right_aligned[sunset_right_aligned.index.date ==
date(2022, 3, 19)]
Expand Down

0 comments on commit fc44c2d

Please sign in to comment.