def test_compute_centroids(self):
initial_centroids = np.array([[3, 3], [6, 2], [8, 5]])
idx = find_closest_centroids(self.X, initial_centroids)
centroids = compute_centroids(self.X, idx, initial_centroids.shape[0])
expected_centroids = np.array([[2.428301, 3.157924],
[5.813503, 2.633656],
[7.119387, 3.616684]])
np.testing.assert_almost_equal(centroids,
expected_centroids,
decimal=6)
def test_compute_centroids(self):
initial_centroids = np.array([[3,3],
[6,2],
[8,5]])
idx = find_closest_centroids(self.X, initial_centroids)
centroids = compute_centroids(self.X, idx, initial_centroids.shape[0])
expected_centroids = np.array([[2.428301, 3.157924],
[5.813503, 2.633656],
[7.119387, 3.616684]])
np.testing.assert_almost_equal(centroids, expected_centroids, decimal=6)
def test_find_closest_centroids(self):
initial_centroids = np.array([[3, 3], [6, 2], [8, 5]])
idx = find_closest_centroids(self.X[0:3, :], initial_centroids)
np.testing.assert_equal(idx, np.array([1, 3, 2]))
def test_find_closest_centroids(self):
initial_centroids = np.array([[3,3],
[6,2],
[8,5]])
idx = find_closest_centroids(self.X[0:3,:], initial_centroids)
np.testing.assert_equal(idx, np.array([1,3,2]))