class TestTableConversion:
'''Test convert to ndarray and DataFrame'''
def setUp(self):
self.headers = ['x', 'y', 'z']
self.t = Table(self.headers)
self.t.append((1,2,3))
self.t.append((4,5,6))
self.t.append((7,8,9))
def test_ndarray(self):
import numpy as np
ta = self.t.as_array()
assert isinstance(ta, np.ndarray)
a = np.array([[1,2,3],
[4,5,6],
[7,8,9]])
assert a.size == ta.size
assert a.shape == ta.shape
assert np.all(a == ta)
assert a is not ta
def test_dataframe_square(self):
import pandas as pd
tp = self.t.as_dataframe()
assert isinstance(tp, pd.DataFrame)
df = pd.DataFrame({'x': [1,4,7],
'y': [2,5,8],
'z': [3,6,9]})
assert len(tp) == len(df)
assert (tp.columns == df.columns).all()
assert (tp == df).all().all()
assert tp is not df
def test_dataframe_nonsquare(self):
import pandas as pd
self.t.append((-1,-1,-1))
tp = self.t.as_dataframe()
assert isinstance(tp, pd.DataFrame)
df = pd.DataFrame({'x': [1,4,7,-1],
'y': [2,5,8,-1],
'z': [3,6,9,-1]})
assert len(tp) == len(df)
assert (tp.columns == df.columns).all()
assert (tp == df).all().all()
assert tp is not df
class TestTableConversion:
'''Test convert to ndarray and DataFrame'''
def setUp(self):
self.headers = ['x', 'y', 'z']
self.t = Table(self.headers)
self.t.append((1, 2, 3))
self.t.append((4, 5, 6))
self.t.append((7, 8, 9))
def test_ndarray(self):
import numpy as np
ta = self.t.as_array()
assert isinstance(ta, np.ndarray)
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
assert a.size == ta.size
assert a.shape == ta.shape
assert np.all(a == ta)
assert a is not ta
def test_dataframe_square(self):
import pandas as pd
tp = self.t.as_dataframe()
assert isinstance(tp, pd.DataFrame)
df = pd.DataFrame({'x': [1, 4, 7], 'y': [2, 5, 8], 'z': [3, 6, 9]})
assert len(tp) == len(df)
assert (tp.columns == df.columns).all()
assert (tp == df).all().all()
assert tp is not df
def test_dataframe_nonsquare(self):
import pandas as pd
self.t.append((-1, -1, -1))
tp = self.t.as_dataframe()
assert isinstance(tp, pd.DataFrame)
df = pd.DataFrame({
'x': [1, 4, 7, -1],
'y': [2, 5, 8, -1],
'z': [3, 6, 9, -1]
})
assert len(tp) == len(df)
assert (tp.columns == df.columns).all()
assert (tp == df).all().all()
assert tp is not df
