Example #1
0
 def testLogical (self):
     "Test logical_and, logical_or, sometrue, alltrue"
     x = MA.array([1,1,0,0])
     y = MA.array([1,0,1,0])
     assert eq(MA.logical_and (x,y), [1,0,0,0])
     assert eq(MA.logical_or (x,y), [1,1,1,0])
     assert MA.sometrue(x)
     assert not MA.sometrue(MA.zeros((3,)))
     assert MA.alltrue(MA.ones((3,)))
     assert not MA.alltrue(x)
Example #2
0
 def testPickle (self):
     "Test pickling of MA arrays."
     import pickle
     x = MA.arange(10)
     fpik = open('foo.pik', 'wb')
     pickle.dump(x, fpik, 0)
     fpik.close()
     fpik = open('foo.pik', 'rb')
     y = pickle.load(open('foo.pik', 'rb'))
     fpik.close()
     assert eq(y,[0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
     assert MA.alltrue(MA.equal(x,y))
     assert MA.sometrue(MA.equal(x,3))
     assert y.shape == (10,)