test_h5types: also test ndarrays

This commit is contained in:
Robert Jördens 2016-04-05 14:58:01 +08:00
parent 7a5b3a1167
commit 8078e59077
1 changed files with 5 additions and 2 deletions

View File

@ -16,8 +16,11 @@ class TypesCase(unittest.TestCase):
} }
for size in 8, 16, 32, 64: for size in 8, 16, 32, 64:
d["i"+str(size)] = getattr(np, "int" + str(size))(42) for typ in "int", "uint":
d["u"+str(size)] = getattr(np, "uint" + str(size))(42) dt = getattr(np, typ + str(size))
d[typ+str(size)] = dt(42)
d["n"+typ+str(size)] = np.array(42, dt)
d["m"+typ+str(size)] = np.array([[[[42]]]], dt)
for size in 16, 32, 64: for size in 16, 32, 64:
d["f"+str(size)] = getattr(np, "float" + str(size))(42) d["f"+str(size)] = getattr(np, "float" + str(size))(42)