def test_adapter_no_rows():
data = DataFrame(columns=['X', 'Y', 'Z'])
viewer = DataFrameViewer(data=data)
adapter = DataFrameAdapter()
item_0_df = adapter.get_item(viewer, 'data', 0)
assert_array_equal(item_0_df.values, np.array([]).reshape(0, 3))
assert_array_equal(item_0_df.columns, ['X', 'Y', 'Z'])
def test_adapter_empty_dataframe():
data = DataFrame()
viewer = DataFrameViewer(data=data)
adapter = DataFrameAdapter()
item_0_df = adapter.get_item(viewer, 'data', 0)
assert_array_equal(item_0_df.values, np.array([]).reshape(0, 0))
assert_array_equal(item_0_df.columns, [])
def test_adapter_get_item():
viewer = sample_data()
adapter = DataFrameAdapter()
item_0_df = adapter.get_item(viewer, 'data', 0)
assert_array_equal(item_0_df.values, [[0, 1, 2]])
assert_array_equal(item_0_df.columns, ['X', 'Y', 'Z'])
assert item_0_df.index[0] == 'one'
def test_adapter_delete_end_numerical_index():
viewer = sample_data_numerical_index()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 3)
data = viewer.data
assert_array_equal(data.values,
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, [1, 2, 3])
def test_adapter_delete_end():
viewer = sample_data()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 3)
data = viewer.data
assert_array_equal(data.values,
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['one', 'two', 'three'])
def test_adapter_delete_middle():
viewer = sample_data()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 1)
data = viewer.data
assert_array_equal(data.values,
[[ 0, 1, 2],
[ 6, 7, 8],
[ 9, 10, 11]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['one', 'three', 'four'])
def test_adapter_delete_middle():
viewer = sample_data()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 1)
data = viewer.data
assert_array_equal(data.values,
[[0, 1, 2],
[6, 7, 8],
[9, 10, 11]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['one', 'three', 'four'])
def test_adapter_delete_start():
viewer = sample_data()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 0)
data = viewer.data
assert_array_equal(data.values,
[[3, 4, 5],
[6, 7, 8],
[9, 10, 11]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['two', 'three', 'four'])
def test_adapter_set_index_text():
viewer = sample_data()
columns = [('', 'index')] + [(column, column)
for column in viewer.data.columns]
adapter = DataFrameAdapter(columns=columns)
adapter.set_text(viewer, 'data', 0, 0, 'NewIndex')
item_0_df = adapter.get_item(viewer, 'data', 0)
assert_array_equal(item_0_df.values, [[0, 1, 2]])
assert_array_equal(item_0_df.columns, ['X', 'Y', 'Z'])
assert item_0_df.index[0] == 'NewIndex'
def test_adapter_delete_end_numerical_index():
viewer = sample_data_numerical_index()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 3)
data = viewer.data
assert_array_equal(data.values,
[[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, [1, 2, 3])
def test_adapter_delete_end():
viewer = sample_data()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 3)
data = viewer.data
assert_array_equal(data.values,
[[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['one', 'two', 'three'])
def test_adapter_delete_middle_numerical_index():
viewer = sample_data_numerical_index()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 1)
data = viewer.data
assert_array_equal(data.values,
[[ 0, 1, 2],
[ 6, 7, 8],
[ 9, 10, 11]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, [1, 3, 4])
def test_adapter_delete_middle_numerical_index():
viewer = sample_data_numerical_index()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 1)
data = viewer.data
assert_array_equal(data.values,
[[0, 1, 2],
[6, 7, 8],
[9, 10, 11]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, [1, 3, 4])
def test_adapter_insert_end_numerical_index():
viewer = sample_data_numerical_index()
adapter = DataFrameAdapter()
item = DataFrame([[-3, -2, -1]], index=[0], columns=['X', 'Y', 'Z'])
adapter.insert(viewer, 'data', 5, item)
data = viewer.data
assert_array_equal(
data.values,
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11], [-3, -2, -1]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, [1, 2, 3, 4, 0])
def test_adapter_insert_end():
viewer = sample_data()
adapter = DataFrameAdapter()
item = DataFrame([[-3, -2, -1]], index=['new'], columns=['X', 'Y', 'Z'])
adapter.insert(viewer, 'data', 5, item)
data = viewer.data
assert_array_equal(
data.values,
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11], [-3, -2, -1]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['one', 'two', 'three', 'four', 'new'])
def test_adapter_delete_start():
viewer = sample_data()
adapter = DataFrameAdapter()
adapter.delete(viewer, 'data', 0)
data = viewer.data
assert_array_equal(data.values,
[[ 3, 4, 5],
[ 6, 7, 8],
[ 9, 10, 11]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['two', 'three', 'four'])
def test_adapter_set_index_text_numeric_invalid(self):
viewer = sample_data_numerical_index()
columns = ([('', 'index')] + [(column, column)
for column in viewer.data.columns])
adapter = DataFrameAdapter(columns=columns)
adapter.set_text(viewer, 'data', 0, 0, 'invalid')
item_0_df = adapter.get_item(viewer, 'data', 0)
assert_array_equal(item_0_df.values, [[0, 1, 2]])
assert_array_equal(item_0_df.columns, ['X', 'Y', 'Z'])
self.assertEqual(item_0_df.index[0], 1)
def test_adapter_insert_middle_numerical_index(self):
viewer = sample_data_numerical_index()
adapter = DataFrameAdapter()
item = DataFrame([[-3, -2, -1]], index=[0], columns=["X", "Y", "Z"])
adapter.insert(viewer, "data", 1, item)
data = viewer.data
assert_array_equal(
data.values,
[[0, 1, 2], [-3, -2, -1], [3, 4, 5], [6, 7, 8], [9, 10, 11]],
)
assert_array_equal(data.columns, ["X", "Y", "Z"])
assert_array_equal(data.index, [1, 0, 2, 3, 4])
def test_adapter_insert_end():
viewer = sample_data()
adapter = DataFrameAdapter()
item = DataFrame([[-3, -2, -1]], index=["new"], columns=["X", "Y", "Z"])
adapter.insert(viewer, "data", 5, item)
data = viewer.data
assert_array_equal(
data.values,
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11], [-3, -2, -1]],
)
assert_array_equal(data.columns, ["X", "Y", "Z"])
assert_array_equal(data.index, ["one", "two", "three", "four", "new"])
def test_adapter_insert_end():
viewer = sample_data()
adapter = DataFrameAdapter()
item = DataFrame([[-3, -2, -1]], index=['new'], columns=['X', 'Y', 'Z'])
adapter.insert(viewer, 'data', 5, item)
data = viewer.data
assert_array_equal(data.values,
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8],
[9, 10, 11],
[-3, -2, -1]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, ['one', 'two', 'three', 'four', 'new'])
def test_adapter_insert_end_numerical_index():
viewer = sample_data_numerical_index()
adapter = DataFrameAdapter()
item = DataFrame([[-3, -2, -1]], index=[0], columns=['X', 'Y', 'Z'])
adapter.insert(viewer, 'data', 5, item)
data = viewer.data
assert_array_equal(data.values,
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8],
[9, 10, 11],
[-3, -2, -1]])
assert_array_equal(data.columns, ['X', 'Y', 'Z'])
assert_array_equal(data.index, [1, 2, 3, 4, 0])
