Example #1
0
 def testSort (self):
     "Test sort, argsort, argmax, argmin"
     s = (3,2,5,1,4,0)
     sm = [s, MA.array(s)[::-1]]
     se = MA.array(s)[0:0]
     assert eq(MA.sort(s), self.a)
     assert len(MA.sort(se)) == 0
     assert eq(MA.argsort(s), [5,3,1,0,4,2])
     assert len(MA.argsort(se)) == 0
     assert eq(MA.sort(sm, axis = -1), [[0,1,2,3,4,5],[0,1,2,3,4,5]])
     assert eq(MA.sort(sm, axis = 0), [[0,2,1,1,2,0],[3,4,5,5,4,3]])
     assert MA.argmax(s) == 2
     assert MA.argmin(s) == 5
     assert eq(MA.argmax(sm, axis=-1), [2,3])
     assert eq(MA.argmax(sm, axis=1), [2,3])
     assert eq(MA.argmax(sm, axis=0), [0,1,0,1,0,1])
     assert eq(MA.argmin(sm, axis=-1), [5,0])
     assert eq(MA.argmin(sm, axis=1), [5,0])