Method test_shape

tests/transforms/test_morphological_ops.py:81–83

(self, input_data, expected_result)

class TestMorph(unittest.TestCase):
    @parameterized.expand(TESTS_SHAPE)
    def test_shape(self, input_data, expected_result):
        result1 = erode(input_data["mask"], input_data["filter_size"])
        assert_allclose(result1.shape, expected_result, type_test=False, device_test=False, atol=0.0)

    @parameterized.expand(TESTS_VALUE_T)
    def test_value_t(self, input_data, expected_result):