def test_generate_with_centers(self):
data = data_generator(3, 1, [5, 10, 15],
[[0.0], [-5.0], [5.0]]).generate()
assertion.eq(30, len(data))
self.assert_distribution(data, [5, 10, 15], [[0.0], [-5.0], [5.0]],
[1.0, 1.0, 1.0])
def test_generate_similar_clusters(self):
data = data_generator(10, 2, 10).generate()
assertion.eq(100, len(data))
self.assert_dimension(data, 2)
def test_generate_two_dimension(self):
data = data_generator(2, 2, [10, 15]).generate()
assertion.eq(25, len(data))
self.assert_dimension(data, 2)
def test_generate_one_cluster(self):
data = data_generator(1, 10, 20).generate()
assertion.eq(20, len(data))
self.assert_dimension(data, 10)
def test_generate_one_dimension(self):
data = data_generator(2, 1, [10, 10]).generate()
assertion.eq(20, len(data))
self.assert_dimension(data, 1)
