diff --git a/tests/attributes_test.py b/tests/attributes_test.py index 621bbf9..7a8e57c 100644 --- a/tests/attributes_test.py +++ b/tests/attributes_test.py @@ -10,6 +10,7 @@ ((False, True), ((2, 2), (1, 3)), torch.randn([4, 4])), ((False, True), ((2, 0), (1, 3)), torch.randn([2, 4])), ((True, False), ((0, 2), (1, 3)), torch.randn([2, 4])), + ((True, False), ((0, 0), (1, 3)), torch.randn([0, 4])), ((True,), ((2, 0),), torch.randn([2])), ((False,), ((0, 2),), torch.randn([2])), ((), (), torch.randn([])),