Unit test coverage for new properties

pytorch · Jan 27, 2025 · 8c283df · 8c283df
1 parent 452597d
commit 8c283df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/models/test_deterministic.py b/test/models/test_deterministic.py
@@ -152,6 +152,8 @@ def test_PosteriorMeanModel(self):
         train_Y = torch.rand(2, 2)
         model = SingleTaskGP(train_X=train_X, train_Y=train_Y)
         mean_model = PosteriorMeanModel(model=model)
+        self.assertTrue(mean_model.num_outputs == train_Y.shape[-1])
+        self.assertTrue(mean_model.batch_shape == torch.Size([]))
 
         test_X = torch.rand(2, 3)
         post = model.posterior(test_X)