@@ -89,30 +89,49 @@ def test_randu_invalid_shape_type() -> None:
8989 # Test with an invalid shape (non-tuple)
9090 with pytest .raises (ValueError ):
9191 shape = [5 , 5 ]
92+ < << << << HEAD
9293 af .randu (shape ) # type: ignore[arg-type]
94+ == == == =
95+ random .randu (shape ) # type: ignore[arg-type]
96+ >> >> >> > 951 aee8 (Update tests . Add randn )
9397
9498
9599# Test cases for the randn function
96100
97101
98102def test_randn_shape_1d () -> None :
99103 shape = (10 ,)
104+ < << << << HEAD
100105 result : af .Array = af .randn (shape )
101106 assert isinstance (result , af .Array )
107+ == == == =
108+ result : Array = random .randn (shape )
109+ assert isinstance (result , Array )
110+ >> >> >> > 951 aee8 (Update tests . Add randn )
102111 assert result .shape == shape
103112
104113
105114def test_randn_shape_2d () -> None :
106115 shape = (5 , 7 )
116+ < << << << HEAD
107117 result : af .Array = af .randn (shape )
108118 assert isinstance (result , af .Array )
119+ == == == =
120+ result : Array = random .randn (shape )
121+ assert isinstance (result , Array )
122+ >> >> >> > 951 aee8 (Update tests . Add randn )
109123 assert result .shape == shape
110124
111125
112126def test_randn_shape_3d () -> None :
113127 shape = (3 , 4 , 6 )
128+ < << << << HEAD
114129 result : af .Array = af .randn (shape )
115130 assert isinstance (result , af .Array )
131+ == == == =
132+ result : Array = random .randn (shape )
133+ assert isinstance (result , Array )
134+ >> >> >> > 951 aee8 (Update tests . Add randn )
116135 assert result .shape == shape
117136
118137
0 commit comments