def test_numpy_loading(self): """Numpy ndarray objects are correctly loaded""" test_data = np.random.randn(6, 3) index = range(test_data.shape[0]) columns = ['a', 'b', 'c'] data = Data.from_numpy(test_data, name='name', columns=columns) ikey = Data._default_index_key expected_values = [{ ikey: i, 'a': row[0], 'b': row[1], 'c': row[2] } for i, row in zip(index, test_data.tolist())] nt.assert_list_equal(expected_values, data.values) nt.assert_equal('name', data.name) index_key = 'akey' data = Data.from_numpy(test_data, name='name', columns=columns, index_key=index_key) expected_values = [{ index_key: i, 'a': row[0], 'b': row[1], 'c': row[2] } for i, row in zip(index, test_data.tolist())] nt.assert_list_equal(expected_values, data.values) index = ['a', 'b', 'c', 'd', 'e', 'f'] data = Data.from_numpy(test_data, name='name', index=index, columns=columns) expected_values = [{ ikey: i, 'a': row[0], 'b': row[1], 'c': row[2] } for i, row in zip(index, test_data.tolist())] nt.assert_list_equal(expected_values, data.values) #Bad loads with nt.assert_raises(LoadError) as err: Data.from_numpy(test_data, 'test', columns, index=range(4)) nt.assert_equal(err.expected, LoadError) columns = ['a', 'b'] with nt.assert_raises(LoadError) as err: Data.from_numpy(test_data, 'test', columns, index) nt.assert_equal(err.expected, LoadError)
def test_numpy_loading(self): """Numpy ndarray objects are correctly loaded""" test_data = np.random.randn(6, 3) index = xrange(test_data.shape[0]) columns = ['a', 'b', 'c'] data = Data.from_numpy(test_data, name='name', columns=columns) ikey = Data._default_index_key expected_values = [ {ikey: i, 'a': row[0], 'b': row[1], 'c': row[2]} for i, row in zip(index, test_data.tolist())] nt.assert_list_equal(expected_values, data.values) nt.assert_equal('name', data.name) index_key = 'akey' data = Data.from_numpy(test_data, name='name', columns=columns, index_key=index_key) expected_values = [ {index_key: i, 'a': row[0], 'b': row[1], 'c': row[2]} for i, row in zip(index, test_data.tolist())] nt.assert_list_equal(expected_values, data.values) index = ['a', 'b', 'c', 'd', 'e', 'f'] data = Data.from_numpy(test_data, name='name', index=index, columns=columns) expected_values = [ {ikey: i, 'a': row[0], 'b': row[1], 'c': row[2]} for i, row in zip(index, test_data.tolist())] nt.assert_list_equal(expected_values, data.values) #Bad loads with nt.assert_raises(LoadError) as err: Data.from_numpy(test_data, 'test', columns, index=xrange(4)) nt.assert_equal(err.expected, LoadError) columns = ['a', 'b'] with nt.assert_raises(LoadError) as err: Data.from_numpy(test_data, 'test', columns, index) nt.assert_equal(err.expected, LoadError)