def test_bug35():
"Bug 35"
txt_array = DataArray(['a','b'], axes=['dummy'])
#calling datarray_to_string on string arrays used to fail
print_grid.datarray_to_string(txt_array)
#because get_formatter returned the class not an instance
assert isinstance(print_grid.get_formatter(txt_array),
print_grid.StrFormatter)
def test_bug35():
"Bug 35"
txt_array = DataArray(['a', 'b'], axes=['dummy'])
#calling datarray_to_string on string arrays used to fail
print_grid.datarray_to_string(txt_array)
#because get_formatter returned the class not an instance
assert isinstance(print_grid.get_formatter(txt_array),
print_grid.StrFormatter)
def test_bug34():
"Bug 34: datetime.date ticks not handled by datarray_to_string"
from datarray.print_grid import datarray_to_string
from datetime import date as D
A = DataArray([[1,2],[3,4]], [('row', ('a', D(2010,1,1))),('col', 'cd')])
exp_out = """row col
--------- -------------------
c d
a 1 2
2010-01-0 3 4"""
nt.assert_equal(datarray_to_string(A), exp_out)
# Output for unsigned integers
B = A.astype(np.uint32)
nt.assert_equal(datarray_to_string(B), exp_out)
def test_bug34():
"Bug 34: datetime.date ticks not handled by datarray_to_string"
from datarray.print_grid import datarray_to_string
from datetime import date as D
A = DataArray([[1,2],[3,4]], [('row', ('a', D(2010,1,1))),('col', 'cd')])
nt.assert_equal(datarray_to_string(A), """row col
--------- -------------------
c d
a 1 2
2010-01-0 3 4""")
def test_bug34():
"Bug 34: datetime.date ticks not handled by datarray_to_string"
from datarray.print_grid import datarray_to_string
from datetime import date as D
A = DataArray([[1, 2], [3, 4]], [('row', ('a', D(2010, 1, 1))),
('col', 'cd')])
nt.assert_equal(
datarray_to_string(A), """row col
--------- -------------------
c d
a 1 2
2010-01-0 3 4""")
def test_1d_datarray_to_string():
grid_string = """
country
---------------------------------------
Netherla Uruguay Germany Spain
0. 0.714286 1.428571 2.142857
""".strip()
test_array = np.arange(20).reshape((4, 5)) / 7.0
row_spec = "country", ["Netherlands", "Uruguay", "Germany", "Spain"]
col_spec = "year", map(str, [1994, 1998, 2002, 2006, 2010])
d_arr = DataArray(test_array, [row_spec, col_spec])
assert datarray_to_string(d_arr.axes.year["1994"]) == grid_string
def test_1d_datarray_to_string():
grid_string = """
country
---------------------------------------
Netherla Uruguay Germany Spain
0. 0.714286 1.428571 2.142857
""".strip()
test_array = np.arange(20).reshape((4, 5)) / 7.0
row_spec = 'country', ['Netherlands', 'Uruguay', 'Germany', 'Spain']
col_spec = 'year', list(map(str, [1994, 1998, 2002, 2006, 2010]))
d_arr = DataArray(test_array, [row_spec, col_spec])
assert datarray_to_string(d_arr.axes.year['1994']) == grid_string
def test_1d_datarray_to_string():
grid_string = """
country
---------------------------------------
Netherla Uruguay Germany Spain
0. 0.714286 1.428571 2.142857
""".strip()
test_array = np.arange(20).reshape((4, 5)) / 7.0
row_spec = 'country', ['Netherlands', 'Uruguay', 'Germany', 'Spain']
col_spec = 'year', list(map(str, [1994, 1998, 2002, 2006, 2010]))
d_arr = DataArray(test_array, [row_spec, col_spec])
assert datarray_to_string(d_arr.axes.year['1994']) == grid_string
def test_2d_datarray_to_string():
grid_string = """
country year
--------- -------------------------------------------------
1994 1998 2002 2006 2010
Netherlan 0. 0.142857 0.285714 0.428571 0.571429
Uruguay 0.714286 0.857143 1. 1.142857 1.285714
Germany 1.428571 1.571429 1.714286 1.857143 2.
Spain 2.142857 2.285714 2.428571 2.571429 2.714286
""".strip()
test_array = np.arange(20).reshape((4, 5)) / 7.0
row_spec = "country", ["Netherlands", "Uruguay", "Germany", "Spain"]
col_spec = "year", map(str, [1994, 1998, 2002, 2006, 2010])
d_arr = DataArray(test_array, [row_spec, col_spec])
assert datarray_to_string(d_arr) == grid_string
def test_2d_datarray_to_string():
grid_string = """
country year
--------- -------------------------------------------------
1994 1998 2002 2006 2010
Netherlan 0. 0.142857 0.285714 0.428571 0.571429
Uruguay 0.714286 0.857143 1. 1.142857 1.285714
Germany 1.428571 1.571429 1.714286 1.857143 2.
Spain 2.142857 2.285714 2.428571 2.571429 2.714286
""".strip()
test_array = np.arange(20).reshape((4, 5)) / 7.0
row_spec = 'country', ['Netherlands', 'Uruguay', 'Germany', 'Spain']
col_spec = 'year', list(map(str, [1994, 1998, 2002, 2006, 2010]))
d_arr = DataArray(test_array, [row_spec, col_spec])
assert datarray_to_string(d_arr) == grid_string
