MCPcopy Create free account
hub / github.com/numpy/numpy / test_get

Method test_get

numpy/ma/tests/test_mrecords.py:48–86  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

46 assert_equal_records(mbase.view(mrecarray), mbase)
47
48 def test_get(self):
49 # Tests fields retrieval
50 base = self.base.copy()
51 mbase = base.view(mrecarray)
52 # As fields..........
53 for field in ('a', 'b', 'c'):
54 assert_equal(getattr(mbase, field), mbase[field])
55 assert_equal(base[field], mbase[field])
56 # as elements .......
57 mbase_first = mbase[0]
58 assert_(isinstance(mbase_first, mrecarray))
59 assert_equal(mbase_first.dtype, mbase.dtype)
60 assert_equal(mbase_first.tolist(), (1, 1.1, b'one'))
61 # Used to be mask, now it's recordmask
62 assert_equal(mbase_first.recordmask, nomask)
63 assert_equal(mbase_first._mask.item(), (False, False, False))
64 assert_equal(mbase_first['a'], mbase['a'][0])
65 mbase_last = mbase[-1]
66 assert_(isinstance(mbase_last, mrecarray))
67 assert_equal(mbase_last.dtype, mbase.dtype)
68 assert_equal(mbase_last.tolist(), (None, None, None))
69 # Used to be mask, now it's recordmask
70 assert_equal(mbase_last.recordmask, True)
71 assert_equal(mbase_last._mask.item(), (True, True, True))
72 assert_equal(mbase_last['a'], mbase['a'][-1])
73 assert_((mbase_last['a'] is masked))
74 # as slice ..........
75 mbase_sl = mbase[:2]
76 assert_(isinstance(mbase_sl, mrecarray))
77 assert_equal(mbase_sl.dtype, mbase.dtype)
78 # Used to be mask, now it's recordmask
79 assert_equal(mbase_sl.recordmask, [0, 1])
80 assert_equal_records(mbase_sl.mask,
81 np.array([(False, False, False),
82 (True, True, True)],
83 dtype=mbase._mask.dtype))
84 assert_equal_records(mbase_sl, base[:2].view(mrecarray))
85 for field in ('a', 'b', 'c'):
86 assert_equal(getattr(mbase_sl, field), base[:2][field])
87
88 def test_set_fields(self):
89 # Tests setting fields.

Callers

nothing calls this directly

Calls 7

assert_equalFunction · 0.90
assert_equal_recordsFunction · 0.90
itemMethod · 0.80
assert_Function · 0.50
copyMethod · 0.45
viewMethod · 0.45
tolistMethod · 0.45

Tested by

no test coverage detected