def test_random(self):
x_grid, y_grid, x_data, y_data, rgb_data = self.random_data(n_grid=50,
n_colors=5,
n_data=121)
img = compute_density(x_grid, y_grid, x_data, y_data, rgb_data)
self.assertTrue(img.shape == (50, 50, 4))
self.assertTrue(np.all(0 <= img) and np.all(img < 256))
def test_single_class(self):
x_grid, y_grid, x_data, y_data, rgb_data = self.random_data(n_grid=50, n_colors=1, n_data=100)
img = compute_density(x_grid, y_grid, x_data, y_data, rgb_data)
self.assertTrue(np.all(img[:, :, 3] == 128))
def test_random(self):
x_grid, y_grid, x_data, y_data, rgb_data = self.random_data(n_grid=50, n_colors=5, n_data=121)
img = compute_density(x_grid, y_grid, x_data, y_data, rgb_data)
self.assertTrue(img.shape == (50, 50, 4))
self.assertTrue(np.all(0 <= img) and np.all(img < 256))
def test_single_class(self):
x_grid, y_grid, x_data, y_data, rgb_data = \
self.random_data(n_grid=50, n_colors=1, n_data=100)
img = compute_density(x_grid, y_grid, x_data, y_data, rgb_data)
self.assertTrue(np.all(img[:, :, 3] == 128))
