summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_recfunctions.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/lib/tests/test_recfunctions.py')
-rw-r--r--numpy/lib/tests/test_recfunctions.py381
1 files changed, 190 insertions, 191 deletions
diff --git a/numpy/lib/tests/test_recfunctions.py b/numpy/lib/tests/test_recfunctions.py
index b175bcb64..d0eda50ce 100644
--- a/numpy/lib/tests/test_recfunctions.py
+++ b/numpy/lib/tests/test_recfunctions.py
@@ -13,24 +13,23 @@ get_names = np.lib.recfunctions.get_names
get_names_flat = np.lib.recfunctions.get_names_flat
zip_descr = np.lib.recfunctions.zip_descr
+
class TestRecFunctions(TestCase):
- """
- Misc tests
- """
- #
+ # Misc tests
+
def setUp(self):
x = np.array([1, 2, ])
y = np.array([10, 20, 30])
z = np.array([('A', 1.), ('B', 2.)],
dtype=[('A', '|S3'), ('B', float)])
w = np.array([(1, (2, 3.0)), (4, (5, 6.0))],
- dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
+ dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
self.data = (w, x, y, z)
-
def test_zip_descr(self):
- "Test zip_descr"
+ # Test zip_descr
(w, x, y, z) = self.data
+
# Std array
test = zip_descr((x, x), flatten=True)
assert_equal(test,
@@ -38,6 +37,7 @@ class TestRecFunctions(TestCase):
test = zip_descr((x, x), flatten=False)
assert_equal(test,
np.dtype([('', int), ('', int)]))
+
# Std & flexible-dtype
test = zip_descr((x, z), flatten=True)
assert_equal(test,
@@ -46,6 +46,7 @@ class TestRecFunctions(TestCase):
assert_equal(test,
np.dtype([('', int),
('', [('A', '|S3'), ('B', float)])]))
+
# Standard & nested dtype
test = zip_descr((x, w), flatten=True)
assert_equal(test,
@@ -58,78 +59,80 @@ class TestRecFunctions(TestCase):
('', [('a', int),
('b', [('ba', float), ('bb', int)])])]))
-
def test_drop_fields(self):
- "Test drop_fields"
+ # Test drop_fields
a = np.array([(1, (2, 3.0)), (4, (5, 6.0))],
dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
+
# A basic field
test = drop_fields(a, 'a')
control = np.array([((2, 3.0),), ((5, 6.0),)],
- dtype=[('b', [('ba', float), ('bb', int)])])
+ dtype=[('b', [('ba', float), ('bb', int)])])
assert_equal(test, control)
+
# Another basic field (but nesting two fields)
test = drop_fields(a, 'b')
control = np.array([(1,), (4,)], dtype=[('a', int)])
assert_equal(test, control)
+
# A nested sub-field
test = drop_fields(a, ['ba', ])
control = np.array([(1, (3.0,)), (4, (6.0,))],
- dtype=[('a', int), ('b', [('bb', int)])])
+ dtype=[('a', int), ('b', [('bb', int)])])
assert_equal(test, control)
+
# All the nested sub-field from a field: zap that field
test = drop_fields(a, ['ba', 'bb'])
control = np.array([(1,), (4,)], dtype=[('a', int)])
assert_equal(test, control)
- #
+
test = drop_fields(a, ['a', 'b'])
assert_(test is None)
-
def test_rename_fields(self):
- "Tests rename fields"
+ # Test rename fields
a = np.array([(1, (2, [3.0, 30.])), (4, (5, [6.0, 60.]))],
dtype=[('a', int),
('b', [('ba', float), ('bb', (float, 2))])])
- test = rename_fields(a, {'a':'A', 'bb':'BB'})
+ test = rename_fields(a, {'a': 'A', 'bb': 'BB'})
newdtype = [('A', int), ('b', [('ba', float), ('BB', (float, 2))])]
control = a.view(newdtype)
assert_equal(test.dtype, newdtype)
assert_equal(test, control)
-
def test_get_names(self):
- "Tests get_names"
+ # Test get_names
ndtype = np.dtype([('A', '|S3'), ('B', float)])
test = get_names(ndtype)
assert_equal(test, ('A', 'B'))
- #
+
ndtype = np.dtype([('a', int), ('b', [('ba', float), ('bb', int)])])
test = get_names(ndtype)
assert_equal(test, ('a', ('b', ('ba', 'bb'))))
-
def test_get_names_flat(self):
- "Test get_names_flat"
+ # Test get_names_flat
ndtype = np.dtype([('A', '|S3'), ('B', float)])
test = get_names_flat(ndtype)
assert_equal(test, ('A', 'B'))
- #
+
ndtype = np.dtype([('a', int), ('b', [('ba', float), ('bb', int)])])
test = get_names_flat(ndtype)
assert_equal(test, ('a', 'b', 'ba', 'bb'))
-
def test_get_fieldstructure(self):
- "Test get_fieldstructure"
+ # Test get_fieldstructure
+
# No nested fields
ndtype = np.dtype([('A', '|S3'), ('B', float)])
test = get_fieldstructure(ndtype)
- assert_equal(test, {'A':[], 'B':[]})
+ assert_equal(test, {'A': [], 'B': []})
+
# One 1-nested field
ndtype = np.dtype([('A', int), ('B', [('BA', float), ('BB', '|S1')])])
test = get_fieldstructure(ndtype)
- assert_equal(test, {'A': [], 'B': [], 'BA':['B', ], 'BB':['B']})
+ assert_equal(test, {'A': [], 'B': [], 'BA': ['B', ], 'BB': ['B']})
+
# One 2-nested fields
ndtype = np.dtype([('A', int),
('B', [('BA', int),
@@ -140,48 +143,48 @@ class TestRecFunctions(TestCase):
assert_equal(test, control)
def test_find_duplicates(self):
- "Test find_duplicates"
+ # Test find_duplicates
a = ma.array([(2, (2., 'B')), (1, (2., 'B')), (2, (2., 'B')),
(1, (1., 'B')), (2, (2., 'B')), (2, (2., 'C'))],
- mask=[(0, (0, 0)), (0, (0, 0)), (0, (0, 0)),
- (0, (0, 0)), (1, (0, 0)), (0, (1, 0))],
- dtype=[('A', int), ('B', [('BA', float), ('BB', '|S1')])])
- #
+ mask=[(0, (0, 0)), (0, (0, 0)), (0, (0, 0)),
+ (0, (0, 0)), (1, (0, 0)), (0, (1, 0))],
+ dtype=[('A', int), ('B', [('BA', float), ('BB', '|S1')])])
+
test = find_duplicates(a, ignoremask=False, return_index=True)
control = [0, 2]
assert_equal(sorted(test[-1]), control)
assert_equal(test[0], a[test[-1]])
- #
+
test = find_duplicates(a, key='A', return_index=True)
control = [0, 1, 2, 3, 5]
assert_equal(sorted(test[-1]), control)
assert_equal(test[0], a[test[-1]])
- #
+
test = find_duplicates(a, key='B', return_index=True)
control = [0, 1, 2, 4]
assert_equal(sorted(test[-1]), control)
assert_equal(test[0], a[test[-1]])
- #
+
test = find_duplicates(a, key='BA', return_index=True)
control = [0, 1, 2, 4]
assert_equal(sorted(test[-1]), control)
assert_equal(test[0], a[test[-1]])
- #
+
test = find_duplicates(a, key='BB', return_index=True)
control = [0, 1, 2, 3, 4]
assert_equal(sorted(test[-1]), control)
assert_equal(test[0], a[test[-1]])
def test_find_duplicates_ignoremask(self):
- "Test the ignoremask option of find_duplicates"
+ # Test the ignoremask option of find_duplicates
ndtype = [('a', int)]
a = ma.array([1, 1, 1, 2, 2, 3, 3],
- mask=[0, 0, 1, 0, 0, 0, 1]).view(ndtype)
+ mask=[0, 0, 1, 0, 0, 0, 1]).view(ndtype)
test = find_duplicates(a, ignoremask=True, return_index=True)
control = [0, 1, 3, 4]
assert_equal(sorted(test[-1]), control)
assert_equal(test[0], a[test[-1]])
- #
+
test = find_duplicates(a, ignoremask=False, return_index=True)
control = [0, 1, 2, 3, 4, 6]
assert_equal(sorted(test[-1]), control)
@@ -189,20 +192,18 @@ class TestRecFunctions(TestCase):
class TestRecursiveFillFields(TestCase):
- """
- Test recursive_fill_fields.
- """
+ # Test recursive_fill_fields.
def test_simple_flexible(self):
- "Test recursive_fill_fields on flexible-array"
+ # Test recursive_fill_fields on flexible-array
a = np.array([(1, 10.), (2, 20.)], dtype=[('A', int), ('B', float)])
b = np.zeros((3,), dtype=a.dtype)
test = recursive_fill_fields(a, b)
control = np.array([(1, 10.), (2, 20.), (0, 0.)],
dtype=[('A', int), ('B', float)])
assert_equal(test, control)
- #
+
def test_masked_flexible(self):
- "Test recursive_fill_fields on masked flexible-array"
+ # Test recursive_fill_fields on masked flexible-array
a = ma.array([(1, 10.), (2, 20.)], mask=[(0, 1), (1, 0)],
dtype=[('A', int), ('B', float)])
b = ma.zeros((3,), dtype=a.dtype)
@@ -211,96 +212,95 @@ class TestRecursiveFillFields(TestCase):
mask=[(0, 1), (1, 0), (0, 0)],
dtype=[('A', int), ('B', float)])
assert_equal(test, control)
- #
-
class TestMergeArrays(TestCase):
- """
- Test merge_arrays
- """
+ # Test merge_arrays
+
def setUp(self):
x = np.array([1, 2, ])
y = np.array([10, 20, 30])
- z = np.array([('A', 1.), ('B', 2.)], dtype=[('A', '|S3'), ('B', float)])
- w = np.array([(1, (2, 3.0)), (4, (5, 6.0))],
+ z = np.array(
+ [('A', 1.), ('B', 2.)], dtype=[('A', '|S3'), ('B', float)])
+ w = np.array(
+ [(1, (2, 3.0)), (4, (5, 6.0))],
dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
self.data = (w, x, y, z)
- #
+
def test_solo(self):
- "Test merge_arrays on a single array."
+ # Test merge_arrays on a single array.
(_, x, _, z) = self.data
- #
+
test = merge_arrays(x)
control = np.array([(1,), (2,)], dtype=[('f0', int)])
assert_equal(test, control)
test = merge_arrays((x,))
assert_equal(test, control)
- #
+
test = merge_arrays(z, flatten=False)
assert_equal(test, z)
test = merge_arrays(z, flatten=True)
assert_equal(test, z)
- #
+
def test_solo_w_flatten(self):
- "Test merge_arrays on a single array w & w/o flattening"
+ # Test merge_arrays on a single array w & w/o flattening
w = self.data[0]
test = merge_arrays(w, flatten=False)
assert_equal(test, w)
- #
+
test = merge_arrays(w, flatten=True)
control = np.array([(1, 2, 3.0), (4, 5, 6.0)],
- dtype=[('a', int), ('ba', float), ('bb', int)])
+ dtype=[('a', int), ('ba', float), ('bb', int)])
assert_equal(test, control)
- #
+
def test_standard(self):
- "Test standard & standard"
+ # Test standard & standard
# Test merge arrays
(_, x, y, _) = self.data
test = merge_arrays((x, y), usemask=False)
control = np.array([(1, 10), (2, 20), (-1, 30)],
dtype=[('f0', int), ('f1', int)])
assert_equal(test, control)
- #
+
test = merge_arrays((x, y), usemask=True)
control = ma.array([(1, 10), (2, 20), (-1, 30)],
mask=[(0, 0), (0, 0), (1, 0)],
dtype=[('f0', int), ('f1', int)])
assert_equal(test, control)
assert_equal(test.mask, control.mask)
- #
+
def test_flatten(self):
- "Test standard & flexible"
+ # Test standard & flexible
(_, x, _, z) = self.data
test = merge_arrays((x, z), flatten=True)
control = np.array([(1, 'A', 1.), (2, 'B', 2.)],
dtype=[('f0', int), ('A', '|S3'), ('B', float)])
assert_equal(test, control)
- #
+
test = merge_arrays((x, z), flatten=False)
control = np.array([(1, ('A', 1.)), (2, ('B', 2.))],
dtype=[('f0', int),
('f1', [('A', '|S3'), ('B', float)])])
assert_equal(test, control)
- #
+
def test_flatten_wflexible(self):
- "Test flatten standard & nested"
+ # Test flatten standard & nested
(w, x, _, _) = self.data
test = merge_arrays((x, w), flatten=True)
control = np.array([(1, 1, 2, 3.0), (2, 4, 5, 6.0)],
dtype=[('f0', int),
('a', int), ('ba', float), ('bb', int)])
assert_equal(test, control)
- #
+
test = merge_arrays((x, w), flatten=False)
controldtype = dtype = [('f0', int),
- ('f1', [('a', int),
- ('b', [('ba', float), ('bb', int)])])]
+ ('f1', [('a', int),
+ ('b', [('ba', float), ('bb', int)])])]
control = np.array([(1., (1, (2, 3.0))), (2, (4, (5, 6.0)))],
dtype=controldtype)
- #
+
def test_wmasked_arrays(self):
- "Test merge_arrays masked arrays"
+ # Test merge_arrays masked arrays
(_, x, _, _) = self.data
mx = ma.array([1, 2, 3], mask=[1, 0, 0])
test = merge_arrays((x, mx), usemask=True)
@@ -311,23 +311,23 @@ class TestMergeArrays(TestCase):
test = merge_arrays((x, mx), usemask=True, asrecarray=True)
assert_equal(test, control)
assert_(isinstance(test, MaskedRecords))
- #
+
def test_w_singlefield(self):
- "Test single field"
+ # Test single field
test = merge_arrays((np.array([1, 2]).view([('a', int)]),
np.array([10., 20., 30.])),)
control = ma.array([(1, 10.), (2, 20.), (-1, 30.)],
mask=[(0, 0), (0, 0), (1, 0)],
dtype=[('a', int), ('f1', float)])
assert_equal(test, control)
- #
+
def test_w_shorter_flex(self):
- "Test merge_arrays w/ a shorter flexndarray."
+ # Test merge_arrays w/ a shorter flexndarray.
z = self.data[-1]
test = merge_arrays((z, np.array([10, 20, 30]).view([('C', int)])))
control = np.array([('A', 1., 10), ('B', 2., 20), ('-1', -1, 20)],
dtype=[('A', '|S3'), ('B', float), ('C', int)])
- #
+
def test_singlerecord(self):
(_, x, y, z) = self.data
test = merge_arrays((x[0], y[0], z[0]), usemask=False)
@@ -338,179 +338,177 @@ class TestMergeArrays(TestCase):
assert_equal(test, control)
-
class TestAppendFields(TestCase):
- """
- Test append_fields
- """
+ # Test append_fields
+
def setUp(self):
x = np.array([1, 2, ])
y = np.array([10, 20, 30])
- z = np.array([('A', 1.), ('B', 2.)], dtype=[('A', '|S3'), ('B', float)])
+ z = np.array(
+ [('A', 1.), ('B', 2.)], dtype=[('A', '|S3'), ('B', float)])
w = np.array([(1, (2, 3.0)), (4, (5, 6.0))],
- dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
+ dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
self.data = (w, x, y, z)
- #
+
def test_append_single(self):
- "Test simple case"
+ # Test simple case
(_, x, _, _) = self.data
test = append_fields(x, 'A', data=[10, 20, 30])
control = ma.array([(1, 10), (2, 20), (-1, 30)],
mask=[(0, 0), (0, 0), (1, 0)],
dtype=[('f0', int), ('A', int)],)
assert_equal(test, control)
- #
+
def test_append_double(self):
- "Test simple case"
+ # Test simple case
(_, x, _, _) = self.data
test = append_fields(x, ('A', 'B'), data=[[10, 20, 30], [100, 200]])
control = ma.array([(1, 10, 100), (2, 20, 200), (-1, 30, -1)],
mask=[(0, 0, 0), (0, 0, 0), (1, 0, 1)],
dtype=[('f0', int), ('A', int), ('B', int)],)
assert_equal(test, control)
- #
+
def test_append_on_flex(self):
- "Test append_fields on flexible type arrays"
+ # Test append_fields on flexible type arrays
z = self.data[-1]
test = append_fields(z, 'C', data=[10, 20, 30])
control = ma.array([('A', 1., 10), ('B', 2., 20), (-1, -1., 30)],
mask=[(0, 0, 0), (0, 0, 0), (1, 1, 0)],
dtype=[('A', '|S3'), ('B', float), ('C', int)],)
assert_equal(test, control)
- #
+
def test_append_on_nested(self):
- "Test append_fields on nested fields"
+ # Test append_fields on nested fields
w = self.data[0]
test = append_fields(w, 'C', data=[10, 20, 30])
control = ma.array([(1, (2, 3.0), 10),
(4, (5, 6.0), 20),
(-1, (-1, -1.), 30)],
- mask=[(0, (0, 0), 0), (0, (0, 0), 0), (1, (1, 1), 0)],
- dtype=[('a', int),
- ('b', [('ba', float), ('bb', int)]),
- ('C', int)],)
+ mask=[(
+ 0, (0, 0), 0), (0, (0, 0), 0), (1, (1, 1), 0)],
+ dtype=[('a', int),
+ ('b', [('ba', float), ('bb', int)]),
+ ('C', int)],)
assert_equal(test, control)
-
class TestStackArrays(TestCase):
- """
- Test stack_arrays
- """
+ # Test stack_arrays
def setUp(self):
x = np.array([1, 2, ])
y = np.array([10, 20, 30])
- z = np.array([('A', 1.), ('B', 2.)], dtype=[('A', '|S3'), ('B', float)])
+ z = np.array(
+ [('A', 1.), ('B', 2.)], dtype=[('A', '|S3'), ('B', float)])
w = np.array([(1, (2, 3.0)), (4, (5, 6.0))],
- dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
+ dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
self.data = (w, x, y, z)
- #
+
def test_solo(self):
- "Test stack_arrays on single arrays"
+ # Test stack_arrays on single arrays
(_, x, _, _) = self.data
test = stack_arrays((x,))
assert_equal(test, x)
self.assertTrue(test is x)
- #
+
test = stack_arrays(x)
assert_equal(test, x)
self.assertTrue(test is x)
- #
+
def test_unnamed_fields(self):
- "Tests combinations of arrays w/o named fields"
+ # Tests combinations of arrays w/o named fields
(_, x, y, _) = self.data
- #
+
test = stack_arrays((x, x), usemask=False)
control = np.array([1, 2, 1, 2])
assert_equal(test, control)
- #
+
test = stack_arrays((x, y), usemask=False)
control = np.array([1, 2, 10, 20, 30])
assert_equal(test, control)
- #
+
test = stack_arrays((y, x), usemask=False)
control = np.array([10, 20, 30, 1, 2])
assert_equal(test, control)
- #
+
def test_unnamed_and_named_fields(self):
- "Test combination of arrays w/ & w/o named fields"
+ # Test combination of arrays w/ & w/o named fields
(_, x, _, z) = self.data
- #
+
test = stack_arrays((x, z))
control = ma.array([(1, -1, -1), (2, -1, -1),
(-1, 'A', 1), (-1, 'B', 2)],
- mask=[(0, 1, 1), (0, 1, 1),
- (1, 0, 0), (1, 0, 0)],
- dtype=[('f0', int), ('A', '|S3'), ('B', float)])
+ mask=[(0, 1, 1), (0, 1, 1),
+ (1, 0, 0), (1, 0, 0)],
+ dtype=[('f0', int), ('A', '|S3'), ('B', float)])
assert_equal(test, control)
assert_equal(test.mask, control.mask)
- #
+
test = stack_arrays((z, x))
control = ma.array([('A', 1, -1), ('B', 2, -1),
(-1, -1, 1), (-1, -1, 2), ],
- mask=[(0, 0, 1), (0, 0, 1),
- (1, 1, 0), (1, 1, 0)],
- dtype=[('A', '|S3'), ('B', float), ('f2', int)])
+ mask=[(0, 0, 1), (0, 0, 1),
+ (1, 1, 0), (1, 1, 0)],
+ dtype=[('A', '|S3'), ('B', float), ('f2', int)])
assert_equal(test, control)
assert_equal(test.mask, control.mask)
- #
+
test = stack_arrays((z, z, x))
control = ma.array([('A', 1, -1), ('B', 2, -1),
('A', 1, -1), ('B', 2, -1),
(-1, -1, 1), (-1, -1, 2), ],
- mask=[(0, 0, 1), (0, 0, 1),
- (0, 0, 1), (0, 0, 1),
- (1, 1, 0), (1, 1, 0)],
- dtype=[('A', '|S3'), ('B', float), ('f2', int)])
+ mask=[(0, 0, 1), (0, 0, 1),
+ (0, 0, 1), (0, 0, 1),
+ (1, 1, 0), (1, 1, 0)],
+ dtype=[('A', '|S3'), ('B', float), ('f2', int)])
assert_equal(test, control)
- #
+
def test_matching_named_fields(self):
- "Test combination of arrays w/ matching field names"
+ # Test combination of arrays w/ matching field names
(_, x, _, z) = self.data
zz = np.array([('a', 10., 100.), ('b', 20., 200.), ('c', 30., 300.)],
dtype=[('A', '|S3'), ('B', float), ('C', float)])
test = stack_arrays((z, zz))
control = ma.array([('A', 1, -1), ('B', 2, -1),
- ('a', 10., 100.), ('b', 20., 200.), ('c', 30., 300.)],
- dtype=[('A', '|S3'), ('B', float), ('C', float)],
- mask=[(0, 0, 1), (0, 0, 1),
- (0, 0, 0), (0, 0, 0), (0, 0, 0)])
+ (
+ 'a', 10., 100.), ('b', 20., 200.), ('c', 30., 300.)],
+ dtype=[('A', '|S3'), ('B', float), ('C', float)],
+ mask=[(0, 0, 1), (0, 0, 1),
+ (0, 0, 0), (0, 0, 0), (0, 0, 0)])
assert_equal(test, control)
assert_equal(test.mask, control.mask)
- #
+
test = stack_arrays((z, zz, x))
ndtype = [('A', '|S3'), ('B', float), ('C', float), ('f3', int)]
control = ma.array([('A', 1, -1, -1), ('B', 2, -1, -1),
('a', 10., 100., -1), ('b', 20., 200., -1),
('c', 30., 300., -1),
(-1, -1, -1, 1), (-1, -1, -1, 2)],
- dtype=ndtype,
- mask=[(0, 0, 1, 1), (0, 0, 1, 1),
- (0, 0, 0, 1), (0, 0, 0, 1), (0, 0, 0, 1),
- (1, 1, 1, 0), (1, 1, 1, 0)])
+ dtype=ndtype,
+ mask=[(0, 0, 1, 1), (0, 0, 1, 1),
+ (0, 0, 0, 1), (0, 0, 0, 1), (0, 0, 0, 1),
+ (1, 1, 1, 0), (1, 1, 1, 0)])
assert_equal(test, control)
assert_equal(test.mask, control.mask)
-
def test_defaults(self):
- "Test defaults: no exception raised if keys of defaults are not fields."
+ # Test defaults: no exception raised if keys of defaults are not fields.
(_, _, _, z) = self.data
zz = np.array([('a', 10., 100.), ('b', 20., 200.), ('c', 30., 300.)],
dtype=[('A', '|S3'), ('B', float), ('C', float)])
- defaults = {'A':'???', 'B':-999., 'C':-9999., 'D':-99999.}
+ defaults = {'A': '???', 'B': -999., 'C': -9999., 'D': -99999.}
test = stack_arrays((z, zz), defaults=defaults)
control = ma.array([('A', 1, -9999.), ('B', 2, -9999.),
- ('a', 10., 100.), ('b', 20., 200.), ('c', 30., 300.)],
- dtype=[('A', '|S3'), ('B', float), ('C', float)],
- mask=[(0, 0, 1), (0, 0, 1),
- (0, 0, 0), (0, 0, 0), (0, 0, 0)])
+ (
+ 'a', 10., 100.), ('b', 20., 200.), ('c', 30., 300.)],
+ dtype=[('A', '|S3'), ('B', float), ('C', float)],
+ mask=[(0, 0, 1), (0, 0, 1),
+ (0, 0, 0), (0, 0, 0), (0, 0, 0)])
assert_equal(test, control)
assert_equal(test.data, control.data)
assert_equal(test.mask, control.mask)
-
def test_autoconversion(self):
- "Tests autoconversion"
+ # Tests autoconversion
adtype = [('A', int), ('B', bool), ('C', float)]
a = ma.array([(1, 2, 3)], mask=[(0, 1, 0)], dtype=adtype)
bdtype = [('A', int), ('B', float), ('C', float)]
@@ -527,9 +525,8 @@ class TestStackArrays(TestCase):
else:
raise AssertionError
-
def test_checktitles(self):
- "Test using titles in the field names"
+ # Test using titles in the field names
adtype = [(('a', 'A'), int), (('b', 'B'), bool), (('c', 'C'), float)]
a = ma.array([(1, 2, 3)], mask=[(0, 1, 0)], dtype=adtype)
bdtype = [(('a', 'A'), int), (('b', 'B'), bool), (('c', 'C'), float)]
@@ -544,16 +541,16 @@ class TestStackArrays(TestCase):
class TestJoinBy(TestCase):
def setUp(self):
self.a = np.array(list(zip(np.arange(10), np.arange(50, 60),
- np.arange(100, 110))),
- dtype=[('a', int), ('b', int), ('c', int)])
+ np.arange(100, 110))),
+ dtype=[('a', int), ('b', int), ('c', int)])
self.b = np.array(list(zip(np.arange(5, 15), np.arange(65, 75),
- np.arange(100, 110))),
- dtype=[('a', int), ('b', int), ('d', int)])
- #
+ np.arange(100, 110))),
+ dtype=[('a', int), ('b', int), ('d', int)])
+
def test_inner_join(self):
- "Basic test of join_by"
+ # Basic test of join_by
a, b = self.a, self.b
- #
+
test = join_by('a', a, b, jointype='inner')
control = np.array([(5, 55, 65, 105, 100), (6, 56, 66, 106, 101),
(7, 57, 67, 107, 102), (8, 58, 68, 108, 103),
@@ -564,7 +561,7 @@ class TestJoinBy(TestCase):
def test_join(self):
a, b = self.a, self.b
- #
+
test = join_by(('a', 'b'), a, b)
control = np.array([(5, 55, 105, 100), (6, 56, 106, 101),
(7, 57, 107, 102), (8, 58, 108, 103),
@@ -574,7 +571,7 @@ class TestJoinBy(TestCase):
def test_outer_join(self):
a, b = self.a, self.b
- #
+
test = join_by(('a', 'b'), a, b, 'outer')
control = ma.array([(0, 50, 100, -1), (1, 51, 101, -1),
(2, 52, 102, -1), (3, 53, 103, -1),
@@ -586,52 +583,53 @@ class TestJoinBy(TestCase):
(9, 69, -1, 104), (10, 70, -1, 105),
(11, 71, -1, 106), (12, 72, -1, 107),
(13, 73, -1, 108), (14, 74, -1, 109)],
- mask=[(0, 0, 0, 1), (0, 0, 0, 1),
- (0, 0, 0, 1), (0, 0, 0, 1),
- (0, 0, 0, 1), (0, 0, 0, 1),
- (0, 0, 1, 0), (0, 0, 0, 1),
- (0, 0, 1, 0), (0, 0, 0, 1),
- (0, 0, 1, 0), (0, 0, 0, 1),
- (0, 0, 1, 0), (0, 0, 0, 1),
- (0, 0, 1, 0), (0, 0, 1, 0),
- (0, 0, 1, 0), (0, 0, 1, 0),
- (0, 0, 1, 0), (0, 0, 1, 0)],
- dtype=[('a', int), ('b', int),
- ('c', int), ('d', int)])
+ mask=[(0, 0, 0, 1), (0, 0, 0, 1),
+ (0, 0, 0, 1), (0, 0, 0, 1),
+ (0, 0, 0, 1), (0, 0, 0, 1),
+ (0, 0, 1, 0), (0, 0, 0, 1),
+ (0, 0, 1, 0), (0, 0, 0, 1),
+ (0, 0, 1, 0), (0, 0, 0, 1),
+ (0, 0, 1, 0), (0, 0, 0, 1),
+ (0, 0, 1, 0), (0, 0, 1, 0),
+ (0, 0, 1, 0), (0, 0, 1, 0),
+ (0, 0, 1, 0), (0, 0, 1, 0)],
+ dtype=[('a', int), ('b', int),
+ ('c', int), ('d', int)])
assert_equal(test, control)
def test_leftouter_join(self):
a, b = self.a, self.b
- #
+
test = join_by(('a', 'b'), a, b, 'leftouter')
control = ma.array([(0, 50, 100, -1), (1, 51, 101, -1),
(2, 52, 102, -1), (3, 53, 103, -1),
(4, 54, 104, -1), (5, 55, 105, -1),
(6, 56, 106, -1), (7, 57, 107, -1),
(8, 58, 108, -1), (9, 59, 109, -1)],
- mask=[(0, 0, 0, 1), (0, 0, 0, 1),
- (0, 0, 0, 1), (0, 0, 0, 1),
- (0, 0, 0, 1), (0, 0, 0, 1),
- (0, 0, 0, 1), (0, 0, 0, 1),
- (0, 0, 0, 1), (0, 0, 0, 1)],
- dtype=[('a', int), ('b', int), ('c', int), ('d', int)])
+ mask=[(0, 0, 0, 1), (0, 0, 0, 1),
+ (0, 0, 0, 1), (0, 0, 0, 1),
+ (0, 0, 0, 1), (0, 0, 0, 1),
+ (0, 0, 0, 1), (0, 0, 0, 1),
+ (0, 0, 0, 1), (0, 0, 0, 1)],
+ dtype=[('a', int), ('b', int), ('c', int), ('d', int)])
class TestJoinBy2(TestCase):
@classmethod
def setUp(cls):
cls.a = np.array(list(zip(np.arange(10), np.arange(50, 60),
- np.arange(100, 110))),
- dtype=[('a', int), ('b', int), ('c', int)])
+ np.arange(100, 110))),
+ dtype=[('a', int), ('b', int), ('c', int)])
cls.b = np.array(list(zip(np.arange(10), np.arange(65, 75),
- np.arange(100, 110))),
- dtype=[('a', int), ('b', int), ('d', int)])
+ np.arange(100, 110))),
+ dtype=[('a', int), ('b', int), ('d', int)])
def test_no_r1postfix(self):
- "Basic test of join_by no_r1postfix"
+ # Basic test of join_by no_r1postfix
a, b = self.a, self.b
- test = join_by('a', a, b, r1postfix='', r2postfix='2', jointype='inner')
+ test = join_by(
+ 'a', a, b, r1postfix='', r2postfix='2', jointype='inner')
control = np.array([(0, 50, 65, 100, 100), (1, 51, 66, 101, 101),
(2, 52, 67, 102, 102), (3, 53, 68, 103, 103),
(4, 54, 69, 104, 104), (5, 55, 70, 105, 105),
@@ -641,15 +639,16 @@ class TestJoinBy2(TestCase):
('c', int), ('d', int)])
assert_equal(test, control)
-
def test_no_postfix(self):
- self.assertRaises(ValueError, join_by, 'a', self.a, self.b, r1postfix='', r2postfix='')
+ self.assertRaises(ValueError, join_by, 'a', self.a, self.b,
+ r1postfix='', r2postfix='')
def test_no_r2postfix(self):
- "Basic test of join_by no_r2postfix"
+ # Basic test of join_by no_r2postfix
a, b = self.a, self.b
- test = join_by('a', a, b, r1postfix='1', r2postfix='', jointype='inner')
+ test = join_by(
+ 'a', a, b, r1postfix='1', r2postfix='', jointype='inner')
control = np.array([(0, 50, 65, 100, 100), (1, 51, 66, 101, 101),
(2, 52, 67, 102, 102), (3, 53, 68, 103, 103),
(4, 54, 69, 104, 104), (5, 55, 70, 105, 105),
@@ -661,25 +660,25 @@ class TestJoinBy2(TestCase):
def test_two_keys_two_vars(self):
a = np.array(list(zip(np.tile([10, 11], 5), np.repeat(np.arange(5), 2),
- np.arange(50, 60), np.arange(10, 20))),
- dtype=[('k', int), ('a', int), ('b', int), ('c', int)])
+ np.arange(50, 60), np.arange(10, 20))),
+ dtype=[('k', int), ('a', int), ('b', int), ('c', int)])
b = np.array(list(zip(np.tile([10, 11], 5), np.repeat(np.arange(5), 2),
- np.arange(65, 75), np.arange(0, 10))),
- dtype=[('k', int), ('a', int), ('b', int), ('c', int)])
+ np.arange(65, 75), np.arange(0, 10))),
+ dtype=[('k', int), ('a', int), ('b', int), ('c', int)])
control = np.array([(10, 0, 50, 65, 10, 0), (11, 0, 51, 66, 11, 1),
(10, 1, 52, 67, 12, 2), (11, 1, 53, 68, 13, 3),
(10, 2, 54, 69, 14, 4), (11, 2, 55, 70, 15, 5),
(10, 3, 56, 71, 16, 6), (11, 3, 57, 72, 17, 7),
(10, 4, 58, 73, 18, 8), (11, 4, 59, 74, 19, 9)],
- dtype=[('k', int), ('a', int), ('b1', int),
- ('b2', int), ('c1', int), ('c2', int)])
- test = join_by(['a', 'k'], a, b, r1postfix='1', r2postfix='2', jointype='inner')
+ dtype=[('k', int), ('a', int), ('b1', int),
+ ('b2', int), ('c1', int), ('c2', int)])
+ test = join_by(
+ ['a', 'k'], a, b, r1postfix='1', r2postfix='2', jointype='inner')
assert_equal(test.dtype, control.dtype)
assert_equal(test, control)
-
if __name__ == '__main__':
run_module_suite()
View Lorry Controller Status