@@ -756,20 +756,17 @@ def test_categorical_moment(p, size, expected):
756
756
"mu, cov, size, expected" ,
757
757
[
758
758
(np .ones (1 ), np .identity (1 ), None , np .ones (1 )),
759
- (np .ones (10 ), np .identity (10 ), None , np .ones (10 )),
759
+ (np .ones (3 ), np .identity (3 ), None , np .ones (3 )),
760
760
(np .ones (2 ), np .identity (2 ), 4 , np .ones ((4 , 2 ))),
761
- (np .ones (2 ), np .identity (2 ), (4 , 3 ), np .ones ((4 , 3 , 2 ))),
762
761
(np .ones ((2 , 2 )), np .identity (2 ), None , np .ones ((2 , 2 ))),
763
- (np .ones ((2 , 2 )), np .identity (2 ), 4 , np .ones ((4 , 2 , 2 ))),
764
- (np .ones ((2 , 2 )), np .identity (2 ), (4 , 2 ), np .ones ((4 , 2 , 2 , 2 ))),
765
762
(np .array ([1 , 0 , 3.0 ]), np .identity (3 ), None , np .array ([1 , 0 , 3.0 ])),
766
763
(np .array ([1 , 0 , 3.0 ]), np .identity (3 ), 4 , np .full ((4 , 3 ), [1 , 0 , 3.0 ])),
767
764
(np .array ([1 , 0 , 3.0 ]), np .identity (3 ), (4 , 2 ), np .full ((4 , 2 , 3 ), [1 , 0 , 3.0 ])),
768
765
(
769
766
np .array ([1 , 3.0 ]),
770
767
np .identity (2 ),
771
- ( 4 , 5 ) ,
772
- np .full ((4 , 5 , 2 ), [1 , 3.0 ]),
768
+ 5 ,
769
+ np .full ((5 , 2 ), [1 , 3.0 ]),
773
770
),
774
771
(
775
772
np .array ([1 , 3.0 ]),
0 commit comments