From 12786b0ef900c477dd2b576557b2c2b4f4628b94 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Tue, 6 Sep 2022 08:36:36 -0700 Subject: [PATCH] Fix pytest warnings --- tests/utils/test_tensor_dataclass.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/tests/utils/test_tensor_dataclass.py b/tests/utils/test_tensor_dataclass.py index 2c79f4992..39121e224 100644 --- a/tests/utils/test_tensor_dataclass.py +++ b/tests/utils/test_tensor_dataclass.py @@ -10,19 +10,19 @@ from nerfactory.utils.tensor_dataclass import TensorDataclass @dataclass -class TestNestedClass(TensorDataclass): +class DummyNestedClass(TensorDataclass): """Dummy dataclass""" x: torch.Tensor @dataclass -class TestTensorDataclass(TensorDataclass): +class DummyTensorDataclass(TensorDataclass): """Dummy dataclass""" a: torch.Tensor b: torch.Tensor - c: TestNestedClass = None + c: DummyNestedClass = None def test_init(): @@ -44,19 +44,19 @@ def test_broadcasting(): a = torch.ones((4, 6, 3)) b = torch.ones((6, 2)) - tensor_dataclass = TestTensorDataclass(a=a, b=b) + tensor_dataclass = DummyTensorDataclass(a=a, b=b) assert tensor_dataclass.b.shape == (4, 6, 2) a = torch.ones((4, 6, 3)) b = torch.ones(2) - tensor_dataclass = TestTensorDataclass(a=a, b=b) + tensor_dataclass = DummyTensorDataclass(a=a, b=b) assert tensor_dataclass.b.shape == (4, 6, 2) # Invalid broadcasting a = torch.ones((4, 6, 3)) b = torch.ones((3, 2)) with pytest.raises(RuntimeError): - tensor_dataclass = TestTensorDataclass(a=a, b=b) + tensor_dataclass = DummyTensorDataclass(a=a, b=b) def test_tensor_ops(): @@ -64,7 +64,7 @@ def test_tensor_ops(): a = torch.ones((4, 6, 3)) b = torch.ones((6, 2)) - tensor_dataclass = TestTensorDataclass(a=a, b=b) + tensor_dataclass = DummyTensorDataclass(a=a, b=b) assert tensor_dataclass.shape == (4, 6) assert tensor_dataclass.a.shape == (4, 6, 3) @@ -96,7 +96,7 @@ def test_tensor_ops(): assert tensor_dataclass[0, ...].shape == (6,) assert tensor_dataclass[0, ...].a.shape == (6, 3) - tensor_dataclass = TestTensorDataclass(a=torch.ones((2, 3, 4, 5)), b=torch.ones((4, 5))) + tensor_dataclass = DummyTensorDataclass(a=torch.ones((2, 3, 4, 5)), b=torch.ones((4, 5))) assert tensor_dataclass[0, ...].shape == (3, 4) assert tensor_dataclass[0, ...].a.shape == (3, 4, 5) assert tensor_dataclass[0, ..., 0].shape == (3,) @@ -113,8 +113,8 @@ def test_nested_class(): a = torch.ones((4, 6, 3)) b = torch.ones((6, 2)) - c = TestNestedClass(x=torch.ones(6, 5)) - tensor_dataclass = TestTensorDataclass(a=a, b=b, c=c) + c = DummyNestedClass(x=torch.ones(6, 5)) + tensor_dataclass = DummyTensorDataclass(a=a, b=b, c=c) assert tensor_dataclass.shape == (4, 6) assert tensor_dataclass.a.shape == (4, 6, 3) @@ -143,7 +143,7 @@ def test_nested_class(): def test_iter(): """Test iterating over tensor dataclass""" - tensor_dataclass = TestTensorDataclass(a=torch.ones((3, 4, 5)), b=torch.ones((3, 4, 5))) + tensor_dataclass = DummyTensorDataclass(a=torch.ones((3, 4, 5)), b=torch.ones((3, 4, 5))) for batch in tensor_dataclass: assert batch.shape == (4,) assert batch.a.shape == (4, 5)