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)