@@ -147,7 +147,7 @@ proc makeTupleTest() =
147147
148148proc stringModTest () =
149149 var inputStr = " This is a nice string, isn't it ?"
150- var res : string = jlUnbox [string ](jlCall (" modString" , (jlBox (inputStr)))
150+ var res : string = jlUnbox [string ](jlCall (" modString" , (jlBox (inputStr))))
151151 check inputStr & " This is an amazing string" == res
152152
153153proc printDictTest () =
@@ -258,8 +258,8 @@ proc runArrayArgsTest() =
258258
259259# ## Tensor Args
260260proc tensorSquareMeBaby () =
261- var orig: Tensor [ float64 ]
262- orig = ones [float64 ](3 , 4 , 5 )
261+ let dims = [ 18 , 21 , 33 ]
262+ var orig: Tensor [ float64 ] = ones [float64 ](dims )
263263 var index = 0
264264 for i in orig.mitems:
265265 i = index.float64 / 3.0
@@ -283,14 +283,14 @@ proc tensorSquareMeBaby() =
283283 var rank_ret = ndims (ret)
284284 check rank_ret == 3
285285 var data_ret = dataArray (ret)
286- var tensorData: Tensor [float64 ] = newTensor [float64 ](3 , 4 , 5 )
286+ var tensorData: Tensor [float64 ] = newTensor [float64 ](dims )
287287 copyMem (tensorData.dataArray (), data_ret, len_ret* sizeof (float64 ))
288288 for i, v in enumerate (tensorData):
289289 check v == (i/ 3 )* (i/ 3 )
290290
291291proc tensorMutateMeBaby () =
292- var orig : Tensor [ float64 ]
293- orig = ones [float64 ](4 , 6 , 8 )
292+ let dims = [ 14 , 12 , 10 ]
293+ var orig : Tensor [ float64 ] = ones [float64 ](dims )
294294 var index = 0
295295 for i in orig.mitems:
296296 inc (index)
@@ -305,7 +305,7 @@ proc tensorMutateMeBaby() =
305305 var rank_ret = ndims (ret)
306306 check rank_ret == 3
307307 var data_ret = dataArray (ret)
308- var tensorData: Tensor [float64 ] = newTensor [float64 ](4 , 6 , 8 )
308+ var tensorData: Tensor [float64 ] = newTensor [float64 ](dims )
309309 copyMem (tensorData.dataArray (), data_ret, len_ret* sizeof (float64 ))
310310 check tensorData == orig
311311
@@ -321,8 +321,8 @@ proc tensorBuiltinRot180() =
321321
322322 var d0 = dim (xArray, 0 ).int
323323 var d1 = dim (xArray, 1 ).int
324- check d0 == 4
325- check d1 == 3
324+ check d0 == orig_tensor.shape[ 0 ]
325+ check d1 == orig_tensor.shape[ 1 ]
326326
327327 var ret = toJlArray [float64 ](jlCall (" rot180" , xArray.toJlValue ()))
328328 check not isNil (ret.data)
0 commit comments