def test_tabulate3(self):
t = Tabulator()
box = Box(get_lst3())
index = ['a', 'b']
columns = ['e']
values = 'd'
pt = t.tabulate(box=box, values=values, columns=columns, index=index)
expected = 'e 1 2\na b \n1 1 12 30\n 2 13 31'
self.assertEqual(str(pt), expected)
def test_tabulate2b(self):
t = Tabulator()
box = Box(get_lst2())
index = ['c']
columns = ['a', 'b']
values = 'd'
pt = t.tabulate(box=box, values=values, columns=columns, index=index)
expected = 'a 1 2 \nb 1 2 1 2\nc \n1 12 13 16 19\n2 30 31 34 37'
self.assertEqual(str(pt), expected)
def test_tabulate(self):
t = Tabulator()
box = Box(get_lst())
index = ['a', 'b']
columns = ['c']
values = 'd'
pt = t.tabulate(box=box, values=values, columns=columns, index=index)
expected = 'c 1 2\na b \n1 1 11 14\n 2 13 12\n2 1 15 16\n 2 17 18'
self.assertEqual(str(pt), expected)
def test_tabulate_missing(self):
t = Tabulator()
box = Box(get_lst8())
index = ['a', 'b']
columns = ['c']
values = 'd'
pt = t.tabulate(box=box, values=values, columns=columns, index=index)
expected = '''c 1 2
a b
1 -999 11 12
2 1 15 16
2 17 18'''
self.assertEqual(str(pt), expected)
def test_vector_table(self):
t = Tabulator()
box = Box(get_lst4())
values = 'd'
index = 'c'
index_vals = [1, 2]
df = t.vector_table(box, values, index, index_vals)
expected = '''a 1 2
b 1 2 1 2
c
1 12.0 13.0 16.0 19.0
2 30.0 31.0 34.0 37.0'''
self.assertEqual(expected, str(df))
def test_tabulate_guessed_indices(self):
t = Tabulator()
box = Box(get_lst())
columns = ['c']
values = 'd'
pt = t.tabulate(box=box, values=values, columns=columns)
expected = '''c 1 2
a b
1 1 11 14
2 13 12
2 1 15 16
2 17 18'''
self.assertEqual(str(pt), expected)
def test_vector_table_cols_default_index_uneven(self):
t = Tabulator()
box = Box(get_lst4_uneven())
values = 'd'
index = 'c'
index_vals = None
df = t.vector_table(box, values, index, index_vals, orient='cols')
expected = '''c 1 2 3
a b
1 1 12.0 30.0 NaN
2 13.0 31.0 20.0
2 1 16.0 34.0 NaN
2 19.0 37.0 NaN'''
self.assertEqual(expected, str(df))
def test_vector_table_cols(self):
t = Tabulator()
box = Box(get_lst4())
values = 'd'
index = 'c'
index_vals = [1, 2]
df = t.vector_table(box, values, index, index_vals, orient='cols')
expected = '''c 1 2
a b
1 1 12.0 30.0
2 13.0 31.0
2 1 16.0 34.0
2 19.0 37.0'''
self.assertEqual(expected, str(df))
def test_vector_table_arr_single_entry_interpolated_2D(self):
""" The index is n-by-1, and must be squeezed """
t = Tabulator()
box = Box(get_lst7())
values = 'd'
index = 'c'
components = ['x', 'y', 'z']
df = t.vector_table(box, values, index, components=components)
expected = '''a 1
b 1
d: x y z
c
1 1.0 3.0 5.0
2 2.0 4.0 6.0'''
self.assertEqual(expected, str(df))
def test_vector_table_arr_single_entry_interpolated(self):
t = Tabulator()
box = Box(get_lst6())
values = 'd'
index = 'c'
index_vals = [1, 2]
components = ['x', 'y', 'z']
df = t.vector_table(box, values, index, components=components)
expected = '''a 1
b 1
d: x y z
c
1 1.0 3.0 5.0
2 2.0 4.0 6.0'''
self.assertEqual(expected, str(df))
def test_vector_table_arr(self):
t = Tabulator()
box = Box(get_lst5())
values = 'd'
index = 'c'
index_vals = [1, 2]
components = ['x', 'y', 'z']
df = t.vector_table(box,
values,
index,
index_vals,
components=components)
expected = '''a 1
b 1 2
d: x y z x y z
c
1 1.0 3.0 5.0 7.0 9.0 11.0
2 2.0 4.0 6.0 8.0 10.0 12.0'''
self.assertEqual(expected, str(df))
def test_to_csv_multiindex_cols(self):
t = Tabulator()
box = Box(get_lst4())
values = 'd'
index = 'c'
index_vals = [1, 2]
df = t.vector_table(box, values, index, index_vals)
with tempfile.TemporaryDirectory() as tmpdirname:
fname = os.path.join(tmpdirname, 'test.csv')
to_csv(df, fname)
with open(fname, 'r') as f:
s = f.read()
expected = '''a,1,1,2,2
b,1,2,1,2
c,,,,
1,12.0,13.0,16.0,19.0
2,30.0,31.0,34.0,37.0
'''
self.assertEqual(s, expected)
def test_to_csv_simple(self):
t = Tabulator()
box = Box(get_lst())
index = ['a', 'b']
columns = ['c']
values = 'd'
pt = t.tabulate(box=box, values=values, columns=columns, index=index)
with tempfile.TemporaryDirectory() as tmpdirname:
fname = os.path.join(tmpdirname, 'test.csv')
to_csv(pt, fname)
with open(fname, 'r') as f:
s = f.read()
expected = '''c,,1,2
a,b,,,
1,1,11,14
1,2,13,12
2,1,15,16
2,2,17,18
'''
self.assertEqual(s, expected)
def test_tabulate_store(self):
t = Tabulator()
box = Box(get_lst2_b())
store = Store()
d = store.new('d', components=['x', 'y'])
index = ['a', 'b']
columns = d.label
values = d
pt = t.tabulate(box=box,
values=values,
columns=columns,
index=index,
store=store)
expected = '''d: x y
a b
1 1 12 30
2 13 31
2 1 16 34
2 19 37'''
self.assertEqual(str(pt), expected)
def test_tabulate_store2(self):
# TODO: FIX
t = Tabulator()
box = Box(get_lst2_b())
store = Store()
d = store.new('d', components=['x', 'y'])
index = d.label
columns = ['a', 'b']
values = d
pt = t.tabulate(box=box,
values=values,
columns=columns,
index=index,
store=store)
expected = '''a 1 2
b 1 2 1 2
d:
x 12 13 16 19
y 30 31 34 37'''
self.assertEqual(str(pt), expected)
def test_tabulate_translated(self):
dct = {
'a': Variable('one', 'blah', 'mm'),
'b': Variable('two', 'blah2', 's'),
'c': Variable('three', 'blah2', 'N')
}
a = Aliases(dct)
t = Tabulator()
box = Box(get_lst())
index = ['a', 'b']
columns = ['c']
values = 'd'
pt = t.tabulate(box=box,
values=values,
columns=columns,
index=index,
aliases=a)
expected = 'three [N] 1 2\none [mm] two [s] \n1 1 11 14\n 2 13 12\n2 1 15 16\n 2 17 18'
self.assertEqual(str(pt), expected)
def test_tabulate_store_nested_cols(self):
t = Tabulator()
box = Box(get_lst2_c())
store = Store()
d = store.new('d', components=['x', 'y'])
index = ['a', 'b']
columns = ['c', d.label]
values = d
pt = t.tabulate(box=box,
values=values,
columns=columns,
index=index,
store=store)
expected = '''c 1 2
d: x y x y
a b
1 1 11 21 14 24
2 13 23 12 22
2 1 15 25 16 26
2 17 27 18 28'''
self.assertEqual(str(pt), expected)
def test_tabulate_store_units(self):
t = Tabulator()
store = Store()
d = store.new('d', unit='mm', components=['x', 'y'])
aliases = Aliases({'d': d})
translated = aliases.translate(get_lst2_b())
box = Box(translated)
index = ['a', 'b']
columns = d.label
values = d
pt = t.tabulate(box=box,
values=values,
columns=columns,
index=index,
store=store)
expected = '''d: x [mm] y [mm]
a b
1 1 12 30
2 13 31
2 1 16 34
2 19 37'''
self.assertEqual(str(pt), expected)