fixing some tests

Author: JSabadin

tests/test_augmentations/test_mosaic.py

@@ -16,15 +16,15 @@
 
 def test_mosaic4():
     img = (np.random.rand(HEIGHT, WIDTH, 3) * 255).astype(np.uint8)
-    mosaic = mosaic4([img, img, img, img], HEIGHT, WIDTH)
+    mosaic = mosaic4([img, img, img, img], HEIGHT, WIDTH, 0, 0, 0)
     assert mosaic.shape == (HEIGHT, WIDTH, 3)
 
 
 def test_bbox_mosaic4():
     bbox = (0, 0, WIDTH, HEIGHT)
     for i in range(4):
         mosaic_bbox = bbox_mosaic4(
-            bbox, HEIGHT // 2, WIDTH // 2, i, HEIGHT, WIDTH
+            bbox, HEIGHT // 2, WIDTH // 2, i, HEIGHT, WIDTH, 0, 0
         )
         assert pytest.approx(mosaic_bbox, abs=0.5) == (
             0,
@@ -45,7 +45,7 @@ def test_keypoint_mosaic4():
         ]
     ):
         mosaic_keypoint = keypoint_mosaic4(
-            keypoint, HEIGHT // 2, WIDTH // 2, i, HEIGHT, WIDTH
+            keypoint, HEIGHT // 2, WIDTH // 2, i, HEIGHT, WIDTH, 0, 0
         )
         assert pytest.approx(mosaic_keypoint, abs=0.25) == (w, h, 0, 0)