def test_histogram_validation(self):
X = 'a'
y = 5
with pytest.raises(TypeError):
v = Histogram(dataset_name='Boston Housing Prices')
v.fit(X, y)
X = [1, 2]
y = [1, 2, 3]
with pytest.raises(ValueError):
v = Histogram(dataset_name='Boston Housing Prices')
v.fit(X, y)
def test_histogram_with_X_df_y_list(self, get_regression_data_df_plus):
directory = "./tests/test_visualate/test_figures/histogram/test_11/"
self.cleanup(directory)
X, _, _, y = get_regression_data_df_plus
v = Histogram(dataset_name='Boston Housing Prices')
v.fit(X, list(y))
v.show(directory=directory)
reports = len(os.listdir(directory))
assert reports == 14, "Expected 14 reports. Got %d" % reports
def test_histogram_with_array_no_title_with_dirpath(
self, get_regression_data):
directory = "./tests/test_visualate/test_figures/histogram/test_2/"
self.cleanup(directory)
X, _ = get_regression_data
v = Histogram(dataset_name='Boston Housing Prices')
v.fit(X)
v.show(directory=directory)
reports = len(os.listdir(directory))
assert reports == 14, "Expected 14 reports. Got %d" % reports
def test_histogram_with_1darray_with_title_with_directory(
self, get_regression_data):
directory = "./tests/test_visualate/test_figures/histogram/test_5/"
self.cleanup(directory)
X, _ = get_regression_data
v = Histogram(dataset_name='Boston Housing Prices',
title="Histogrammy5")
v.fit(X[:, 1])
v.show(directory=directory)
reports = len(os.listdir(directory))
assert reports == 1, "Expected 1 reports. Got %d" % reports
def test_histogram_with_array_with_title_with_dirpath(
self, get_regression_data):
kwargs = {'template': 'none'}
directory = "./tests/test_visualate/test_figures/histogram/test_3/"
self.cleanup(directory)
X, _ = get_regression_data
v = Histogram(title="Histogrammy3",
dataset_name='Boston Housing Prices',
**kwargs)
v.fit(X)
v.show(directory=directory)
reports = len(os.listdir(directory))
assert reports == 13, "Expected 13 reports. Got %d" % reports