def test_calculate_average_from_empty_list():
"""
Test to see if it raises an exception
when trying to calculate average form a zero length dataset
"""
# Setup
avg = Averager(3)
# Exercise/Verify
with pytest.raises(IndexError):
avg.running_mean()
def test_average_for_one_data_point():
"""Test calculating average for one data point"""
# Setup
avg = Averager(3)
# Exercise
avg.add_data(5)
average = avg.running_mean()
# Verify
assert average == 5
def test_average_for_2n_data_points():
"""Test calculating average for 2n data point"""
# Setup
avg = Averager(3)
# Exercise
for i in range(1, 7):
avg.add_data(i)
average = avg.running_mean()
# Verify
assert average == 5
def test_average_for_n_plus_one_data_points():
"""Test calculating average for n + 1 data point"""
# Setup
avg = Averager(3)
# Exercise
avg.add_data(5)
avg.add_data(7)
avg.add_data(9)
avg.add_data(11)
average = avg.running_mean()
# Verify
assert average == 9