1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
# pylint: disable-msg=W0611, W0612, W0511,R0201
"""Tests suite for mrecarray.
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_mrecarray.py 68 2007-01-10 17:46:04Z backtopop $
"""
__author__ = "Pierre GF Gerard-Marchant ($Author: backtopop $)"
__version__ = '1.0'
__revision__ = "$Revision: 68 $"
__date__ = '$Date: 2007-01-10 12:46:04 -0500 (Wed, 10 Jan 2007) $'
import types
import numpy as N
import numpy.core.fromnumeric as fromnumeric
from numpy.testing import NumpyTest, NumpyTestCase
from numpy.testing.utils import build_err_msg
import maskedarray.testutils
from maskedarray.testutils import *
import maskedarray.core as MA
import maskedarray.mrecords
from maskedarray.mrecords import mrecarray, fromarrays, fromtextfile, fromrecords
#..............................................................................
class test_mrecarray(NumpyTestCase):
"Base test class for MaskedArrays."
def __init__(self, *args, **kwds):
NumpyTestCase.__init__(self, *args, **kwds)
self.setup()
def setup(self):
"Generic setup"
d = N.arange(5)
m = MA.make_mask([1,0,0,1,1])
base_d = N.r_[d,d[::-1]].reshape(2,-1).T
base_m = N.r_[[m, m[::-1]]].T
base = MA.array(base_d, mask=base_m)
mrec = fromarrays(base.T,)
self.data = [d, m, mrec]
def test_get(self):
"Tests fields retrieval"
[d, m, mrec] = self.data
assert_equal(mrec.f0, MA.array(d,mask=m))
assert_equal(mrec.f1, MA.array(d[::-1],mask=m[::-1]))
assert((mrec._fieldmask == N.core.records.fromarrays([m, m[::-1]])).all())
assert_equal(mrec._mask, N.r_[[m,m[::-1]]].all(0))
assert_equal(mrec.f0, mrec['f0'])
def test_set(self):
"Tests setting fields/attributes."
[d, m, mrec] = self.data
mrec.f0_data = 5
assert_equal(mrec['f0_data'], [5,5,5,5,5])
mrec.f0 = 1
assert_equal(mrec['f0_data'], [1]*5)
assert_equal(mrec['f0_mask'], [0]*5)
mrec.f1 = MA.masked
assert_equal(mrec.f1.mask, [1]*5)
assert_equal(mrec['f1_mask'], [1]*5)
mrec._mask = MA.masked
assert_equal(mrec['f1_mask'], [1]*5)
assert_equal(mrec['f0_mask'],mrec['f1_mask'])
mrec._mask = MA.nomask
assert_equal(mrec['f1_mask'], [0]*5)
assert_equal(mrec['f0_mask'],mrec['f1_mask'])
def test_hardmask(self):
"Test hardmask"
[d, m, mrec] = self.data
print mrec._mask
mrec.harden_mask()
assert(mrec._hardmask)
mrec._mask = nomask
assert_equal(mrec._mask, N.r_[[m,m[::-1]]].all(0))
mrec.soften_mask()
assert(not mrec._hardmask)
mrec._mask = nomask
assert_equal(mrec['f1_mask'], [0]*5)
assert_equal(mrec['f0_mask'],mrec['f1_mask'])
def test_fromtextfile(self):
"Tests reading from a text file."
fcontent = """#
'One (S)','Two (I)','Three (F)','Four (M)','Five (-)','Six (C)'
'strings',1,1.0,'mixed column',,1
'with embedded "double quotes"',2,2.0,1.0,,1
'strings',3,3.0E5,3,,1
'strings',4,-1e-10,,,1
"""
import os
from datetime import datetime
fname = 'tmp%s' % datetime.now().strftime("%y%m%d%H%M%S%s")
f = open(fname, 'w')
f.write(fcontent)
f.close()
mrectxt = fromtextfile(fname,delimitor=',',varnames='ABCDEFG')
os.unlink(fname)
#
assert(isinstance(mrectxt, mrecarray))
assert_equal(mrectxt.F, [1,1,1,1])
assert_equal(mrectxt.E._mask, [1,1,1,1])
assert_equal(mrectxt.C, [1,2,3.e+5,-1e-10])
def test_fromrecords(self):
"Test from recarray."
[d, m, mrec] = self.data
nrec = N.core.records.fromarrays(N.r_[[d,d[::-1]]])
mrecfr = fromrecords(nrec.tolist())
assert_equal(mrecfr.f0, mrec.f0)
assert_equal(mrecfr.dtype, mrec.dtype)
#....................
mrecfr = fromrecords(nrec)
assert_equal(mrecfr.f0, mrec.f0)
assert_equal(mrecfr.dtype, mrec.dtype)
#....................
tmp = mrec[::-1] #.tolist()
mrecfr = fromrecords(tmp)
assert_equal(mrecfr.f0, mrec.f0[::-1])
###############################################################################
#------------------------------------------------------------------------------
if __name__ == "__main__":
NumpyTest().run()
|
