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])