diff --git a/tests/metrics/test_ssim_metric.py b/tests/metrics/test_ssim_metric.py index d79107e999..f1b62f916f 100644 --- a/tests/metrics/test_ssim_metric.py +++ b/tests/metrics/test_ssim_metric.py @@ -21,7 +21,7 @@ class TestSSIMMetric(unittest.TestCase): - def test2d_gaussian(self): + def test_2d_gaussian(self): set_determinism(0) preds = torch.abs(torch.randn(2, 3, 16, 16)) target = torch.abs(torch.randn(2, 3, 16, 16)) @@ -34,7 +34,7 @@ def test2d_gaussian(self): expected_value = 0.045415 self.assertTrue(expected_value - result.item() < 0.000001) - def test2d_uniform(self): + def test_2d_uniform(self): set_determinism(0) preds = torch.abs(torch.randn(2, 3, 16, 16)) target = torch.abs(torch.randn(2, 3, 16, 16)) @@ -47,7 +47,7 @@ def test2d_uniform(self): expected_value = 0.050103 self.assertTrue(expected_value - result.item() < 0.000001) - def test3d_gaussian(self): + def test_3d_gaussian(self): set_determinism(0) preds = torch.abs(torch.randn(2, 3, 16, 16, 16)) target = torch.abs(torch.randn(2, 3, 16, 16, 16))