def test_save_npy_3d():
"""Save 3D array to npy"""
arr = numpy.zeros((20, 10, 3), dtype='float')
arr[10, 5, 1] = 249.49
parameters = {'path': 'array.npy', 'data': [arr]}
assert arrays.save(parameters)
arr = numpy.load('array.npy')
assert_equal(arr[10, 5, 1], 249.49)
assert_equal(arr.sum(), 249.49)
def test_save_npy_3d():
"""Save 3D array to npy"""
arr = numpy.zeros((20, 10, 3), dtype='float')
arr[10, 5, 1] = 249.49
parameters = {'path': 'array.npy', 'data': [arr]}
assert arrays.save(parameters)
arr = numpy.load('array.npy')
assert_equal(arr[10, 5, 1], 249.49)
assert_equal(arr.sum(), 249.49)
def test_save_fail():
"""Fail to save file with unkown extension"""
arr = numpy.zeros((20, 10), dtype='float')
parameters = {'path': 'foo.bar', 'data': [arr]}
arrays.save(parameters)
def test_save_npy():
"""Save 2D array to npy"""
arr = numpy.zeros((20, 10), dtype='float')
parameters = {'path': 'array.npy', 'data': [arr]}
assert arrays.save(parameters)
def test_save_txt_fail():
"""Fail to save 3D array to txt, numpy doesn't support it"""
arr = numpy.zeros((20, 10, 3), dtype='float')
parameters = {'path': 'array.txt', 'data': [arr]}
arrays.save(parameters)
def test_save_fail():
"""Fail to save file with unkown extension"""
arr = numpy.zeros((20, 10), dtype='float')
parameters = {'path': 'foo.bar', 'data': [arr]}
arrays.save(parameters)
def test_save_npy():
"""Save 2D array to npy"""
arr = numpy.zeros((20, 10), dtype='float')
parameters = {'path': 'array.npy', 'data': [arr]}
assert arrays.save(parameters)
def test_save_txt_fail():
"""Fail to save 3D array to txt, numpy doesn't support it"""
arr = numpy.zeros((20, 10, 3), dtype='float')
parameters = {'path': 'array.txt', 'data': [arr]}
arrays.save(parameters)
