add last changes to test folder

Author: Stealeristaken

tests/test_calculations.py

@@ -7,7 +7,7 @@
 def test_get_embeddings():
     image_paths = read_images_from_directory("tests/test_images")
     images = read_with_pil(image_paths)
-    embeddings = get_embeddings(images=images, use_gpu=False)
+    embeddings = get_embeddings(images=images, use_gpu=False, model="resnet-18")
     assert embeddings is not None, "Embeddings were not generated."
 
 
@@ -20,7 +20,7 @@ def test_calculate_pca():
 def test_calculate_kmeans():
     pca_embeddings = np.random.rand(10, 16)
     num_classes = 2
-    centroid, labels, counts = calculate_kmeans(pca_embeddings, num_classes)
+    centroid, labels, counts = calculate_kmeans(pca_embeddings, num_classes, iter=10)
     assert (
         len(set(labels)) == num_classes
     ), "K-means did not cluster into the expected number of classes."