class TestSimpleLinearRegression(object):
def setup(self):
print ("setup() before any methods in this class")
self.hypo = SimpleLinearRegression()
"""
Tests
"""
def test_simple_linear_regression_instance(self):
assert_is_instance(self.hypo, SimpleLinearRegression)
def test_initialize_parameters_correctly_sets_parameter_estimates(self):
self.hypo.initialize_parameters(PARAM_DICT)
assert_equals(self.hypo.intercept.value, INTERCEPT_VALUE)
assert_equals(self.hypo.slope.value, SLOPE_VALUE)
def test_update_parameters_raises_error_without_initialization(self):
assert_raises(ParameterValuesNotInitialized, self.hypo.update_parameters,
PARAM_ARRAY)
def setup(self):
print ("setup() before any methods in this class")
self.hypo = SimpleLinearRegression()
