diff --git a/test/dataset/test_utils.py b/test/dataset/test_utils.py index 0bf8ecb4..e269dde7 100644 --- a/test/dataset/test_utils.py +++ b/test/dataset/test_utils.py @@ -275,19 +275,15 @@ def test_map_detections_class_id( DoesNotRaise(), ), # mask where foreground consists of 3 separate components ( - np.array( - [[[]]] - ).astype(bool), + np.array([[[]]]).astype(bool), None, pytest.raises(AssertionError), - ), # raises AssertionError because mask dimentionality is not 2D + ), # raises AssertionError because mask dimentionality is not 2D ( - np.array( - [[]] - ).astype(bool), + np.array([[]]).astype(bool), None, pytest.raises(AssertionError), - ), # raises AssertionError because mask is empty + ), # raises AssertionError because mask is empty ], ) def test_mask_to_rle_conversion( @@ -346,8 +342,8 @@ def test_mask_to_rle_conversion( [2, 2], None, pytest.raises(AssertionError), - ), # raises AssertionError because number of pixels in RLE does not match - # number of pixels in expected mask (width x height). + ), # raises AssertionError because number of pixels in RLE does not match + # number of pixels in expected mask (width x height). ], ) def test_rle_to_mask_convertion(