def METHOD_NAME(self):
    filter_2d = torch.ones(3, 3)
    image_2d = torch.ones(1, 3, 3)
    apply_filter_2d = ApplyFilter(filter_2d)
    out = apply_filter_2d(image_2d)
    self.assertEqual(out.shape, image_2d.shape)