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_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))