You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
5 years ago
|
""" Testing
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import division, print_function, absolute_import
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
from numpy.testing import assert_array_equal, assert_array_almost_equal, \
|
||
|
|
assert_
|
||
|
|
|
||
|
|
from scipy.io.matlab.mio_utils import squeeze_element, chars_to_strings
|
||
|
|
|
||
|
|
|
||
|
|
def test_squeeze_element():
|
||
|
|
a = np.zeros((1,3))
|
||
|
|
assert_array_equal(np.squeeze(a), squeeze_element(a))
|
||
|
|
# 0d output from squeeze gives scalar
|
||
|
|
sq_int = squeeze_element(np.zeros((1,1), dtype=float))
|
||
|
|
assert_(isinstance(sq_int, float))
|
||
|
|
# Unless it's a structured array
|
||
|
|
sq_sa = squeeze_element(np.zeros((1,1),dtype=[('f1', 'f')]))
|
||
|
|
assert_(isinstance(sq_sa, np.ndarray))
|
||
|
|
|
||
|
|
|
||
|
|
def test_chars_strings():
|
||
|
|
# chars as strings
|
||
|
|
strings = ['learn ', 'python', 'fast ', 'here ']
|
||
|
|
str_arr = np.array(strings, dtype='U6') # shape (4,)
|
||
|
|
chars = [list(s) for s in strings]
|
||
|
|
char_arr = np.array(chars, dtype='U1') # shape (4,6)
|
||
|
|
assert_array_equal(chars_to_strings(char_arr), str_arr)
|
||
|
|
ca2d = char_arr.reshape((2,2,6))
|
||
|
|
sa2d = str_arr.reshape((2,2))
|
||
|
|
assert_array_equal(chars_to_strings(ca2d), sa2d)
|
||
|
|
ca3d = char_arr.reshape((1,2,2,6))
|
||
|
|
sa3d = str_arr.reshape((1,2,2))
|
||
|
|
assert_array_equal(chars_to_strings(ca3d), sa3d)
|
||
|
|
# Fortran ordered arrays
|
||
|
|
char_arrf = np.array(chars, dtype='U1', order='F') # shape (4,6)
|
||
|
|
assert_array_equal(chars_to_strings(char_arrf), str_arr)
|
||
|
|
# empty array
|
||
|
|
arr = np.array([['']], dtype='U1')
|
||
|
|
out_arr = np.array([''], dtype='U1')
|
||
|
|
assert_array_equal(chars_to_strings(arr), out_arr)
|
||
|
|
|