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)
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,)