""" Testing
"""
import numpy as np
from nose.tools import assert_true,assert_false,\
assert_equal, assert_raises
from numpy.testing import assert_array_equal,assert_array_almost_equal
from scipy.io.matlab.mio_utils import cproduct,squeeze_element,\
chars_to_strings
def test_cproduct():
yield assert_equal, cproduct(()), 1
yield assert_equal, cproduct((1,)), 1
yield assert_equal, cproduct((1,3)), 3
yield assert_equal, cproduct([1,3]), 3
def test_squeeze_element():
a = np.zeros((1,3))
yield (assert_array_equal,
np.squeeze(a),
squeeze_element(a))
# 0d output from squeeze gives scalar
sq_int = squeeze_element(np.zeros((1,1), dtype=np.float))
yield assert_true, isinstance(sq_int, float)
# Unless it's a structured array
sq_sa = squeeze_element(np.zeros((1,1),dtype=[('f1', 'f')]))
yield assert_true, 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)
yield assert_array_equal, chars_to_strings(char_arr), str_arr
ca2d = char_arr.reshape((2,2,6))
sa2d = str_arr.reshape((2,2))
yield assert_array_equal, chars_to_strings(ca2d), sa2d
ca3d = char_arr.reshape((1,2,2,6))
sa3d = str_arr.reshape((1,2,2))
yield assert_array_equal, chars_to_strings(ca3d), sa3d
# Fortran ordered arrays
char_arrf = np.array(chars, dtype='U1', order='F') # shape (4,6)
yield assert_array_equal, chars_to_strings(char_arrf), str_arr
# empty array
arr = np.array([['']], dtype='U1')
out_arr = np.array([''], dtype='U1')
yield assert_array_equal, chars_to_strings(arr), out_arr
|