def testListListListFloat(self):
""" Test a float in a list of a list of a list."""
num = [[[1.0]]]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0], dtype=float))
def testListListListFloat(self):
""" Test a float in a list of a list of a list."""
num = [[[1.0]]]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0], dtype=float))
def testTupleAllTypes2(self):
"""Test tuple of a single value for all data types."""
for dtype in [int, int8, int16, int32, int64, float, float16, float32, float64, float128]:
num = (dtype(1), dtype(2))
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1, 2], dtype=float))
def testTupleScalar(self):
"""Test single scalar value in tuple."""
num = (1)
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0], dtype=float))
def testNpArray(self):
"""Test multiple values in numpy array."""
num = np.array([1, 2])
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0, 2.0], dtype=float))
def testTupleScalar(self):
"""Test single scalar value in tuple."""
num = (1)
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0], dtype=float))
def testListAllTypes2(self):
"""List of list of numbers of all data types."""
for dtype in [int, int8, int16, int32, int64, float, float16, float32, float64, float128]:
num = [dtype(1), dtype(2)]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0, 2.0], dtype=float))
def testAllNumpyArrayTypes(self):
"""Test scalar value in numpy array of ndim=0 for all data types."""
for dtype in [int, int8, int16, int32, int64, float, float16, float32, float64, float128]:
num = np.array(1, dtype=dtype)
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0], dtype=float))
def testAllScalarTypes(self):
"""Test single scalar value for all valid data types."""
for dtype in [int, int8, int16, int32, int64, float, float16, float32, float64, float128]:
num = dtype(1)
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0], dtype=float))
def testListListListInts(self):
"""Test 2 lists of ints in a list in a list."""
num = [[[1,1],[2,2]]]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0, 1.0], dtype=float))
np.testing.assert_array_equal(num_[0][1], array([2.0, 2.0], dtype=float))
def testNpArray(self):
"""Test multiple values in numpy array."""
num = np.array([1, 2])
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0, 2.0],
dtype=float))
def testListTupleArray(self):
"""List of tuple of numpy array for all valid types."""
for dtype in int, int8, int16, int32, int64, float, float16, float32, float64, float128:
num = [(array([1,1], dtype=dtype),array([2,2], dtype=dtype))]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0, 1.0], dtype=float))
np.testing.assert_array_equal(num_[0][1], array([2.0, 2.0], dtype=float))
def testListListListInts(self):
"""Test 2 lists of ints in a list in a list."""
num = [[[1, 1], [2, 2]]]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0, 1.0],
dtype=float))
np.testing.assert_array_equal(num_[0][1], array([2.0, 2.0],
dtype=float))
def testTupleAllTypes2(self):
"""Test tuple of a single value for all data types."""
for dtype in [
int, int8, int16, int32, int64, float, float16, float32,
float64, float128
]:
num = (dtype(1), dtype(2))
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1, 2],
dtype=float))
def testListTupleArray(self):
"""List of tuple of numpy array for all valid types."""
for dtype in int, int8, int16, int32, int64, float, float16, float32, float64, float128:
num = [(array([1, 1], dtype=dtype), array([2, 2], dtype=dtype))]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0],
array([1.0, 1.0], dtype=float))
np.testing.assert_array_equal(num_[0][1],
array([2.0, 2.0], dtype=float))
def testListAllTypes2(self):
"""List of list of numbers of all data types."""
for dtype in [
int, int8, int16, int32, int64, float, float16, float32,
float64, float128
]:
num = [dtype(1), dtype(2)]
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0],
array([1.0, 2.0], dtype=float))
def testAllNumpyArrayTypes2(self):
"""Test numpy array for all types."""
for dtype in [
int, int8, int16, int32, int64, float, float16, float32,
float64, float128
]:
num = np.array([1, 2], dtype=dtype)
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0],
array([1.0, 2.0], dtype=float))
def testAllScalarTypes(self):
"""Test single scalar value for all valid data types."""
for dtype in [
int, int8, int16, int32, int64, float, float16, float32,
float64, float128
]:
num = dtype(1)
num_ = _cleanPart(num)
assert isinstance(num_, list)
assert np.all([isinstance(part, list) for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0],
dtype=float))
def testListListArrays(self):
"""List of list of numpy arrays for all valid types."""
for dtype in int, int8, int16, int32, int64, float, float16, float32, float64, float128:
num = [[array([1,1], dtype=dtype),array([2,2], dtype=dtype)],
[array([3,3], dtype=dtype),array([4,4], dtype=dtype)]]
num_ = _cleanPart(num)
assert len(num_) == 2
assert np.all([isinstance(part, list) for part in num_])
assert np.all([len(part) == 2 for part in num_])
np.testing.assert_array_equal(num_[0][0], array([1.0, 1.0], dtype=float))
np.testing.assert_array_equal(num_[0][1], array([2.0, 2.0], dtype=float))
np.testing.assert_array_equal(num_[1][0], array([3.0, 3.0], dtype=float))
np.testing.assert_array_equal(num_[1][1], array([4.0, 4.0], dtype=float))
def testListListArrays(self):
"""List of list of numpy arrays for all valid types."""
for dtype in int, int8, int16, int32, int64, float, float16, float32, float64, float128:
num = [[array([1, 1], dtype=dtype),
array([2, 2], dtype=dtype)],
[array([3, 3], dtype=dtype),
array([4, 4], dtype=dtype)]]
num_ = _cleanPart(num)
assert len(num_) == 2
assert np.all([isinstance(part, list) for part in num_])
assert np.all([len(part) == 2 for part in num_])
np.testing.assert_array_equal(num_[0][0],
array([1.0, 1.0], dtype=float))
np.testing.assert_array_equal(num_[0][1],
array([2.0, 2.0], dtype=float))
np.testing.assert_array_equal(num_[1][0],
array([3.0, 3.0], dtype=float))
np.testing.assert_array_equal(num_[1][1],
array([4.0, 4.0], dtype=float))