def test_is_greater_than_average(): input_series = pd.Series([1, 2, 3, 2.5, 4]) expected_series = pd.Series([0, 0, 1, 0, 1]) output_series = is_greater_than_average(series=input_series) assert_series_equal(output_series, expected_series)
def test_is_greater_than_average_empty(): input_series = pd.Series([]) expected_series = pd.Series([]) actual_series = is_greater_than_average(series=input_series) assert_series_equal(actual_series, expected_series)
def test_is_returning_series(): input_series = pd.Series([1, 2, 3, 2.5, 4]) output_series = is_greater_than_average(series=input_series) type_as_string = type(output_series).__name__ assert type_as_string == "Series"
def test_series_type(): input_series = pd.Series([1, 2, 3, 2.5, 4]) output_series = is_greater_than_average(series=input_series) assert type(output_series) is pd.Series
def test_is_greater_than_average_returns_series(): input_series = pd.Series([1, 2, 3, 2.5, 4]) output_series = is_greater_than_average(series=input_series) assert type(output_series) is pd.core.series.Series
def test_is_greater_than_average_all_the_same(): input_series = pd.Series([10, 10, 10, 10]) expected_series = pd.Series([0, 0, 0, 0]) actual_series = is_greater_than_average(series=input_series) assert_series_equal(actual_series, expected_series)
def test_is_greater_than_average_with_nan(): input_series = pd.Series([1,2,np.nan]) expected_series = pd.Series([0,1,np.nan]) # or exception? actual_series = is_greater_than_average(series=input_series) assert_series_equal(actual_series, expected_series)