def test_copyDataFrame(copy, operator): dataFrame = pandas.DataFrame([0], columns=['A']) model = DataFrameModel(dataFrame, copyDataFrame=copy) assert operator(id(model.dataFrame()), id(dataFrame)) model.setDataFrame(dataFrame, copyDataFrame=copy) assert operator(id(model.dataFrame()), id(dataFrame))
def test_setDataFrame(): dataFrame = pandas.DataFrame([0], columns=['A']) model = DataFrameModel() model.setDataFrame(dataFrame) assert not model.dataFrame().empty assert model.dataFrame() is dataFrame with pytest.raises(TypeError) as excinfo: model.setDataFrame(None) assert "pandas.core.frame.DataFrame" in str(excinfo.value)
def open_merge_purge_dialog(self, model: DataFrameModel = None): if model is None: model = self.get_tree_selected_model() df = model.dataFrame() file_base, ext = os.path.splitext(model.filePath) dedupe_model = create_standard_item_model(df.columns) sort_model = create_standard_item_model(df.columns) self.dialog_merge_purge.set_handler_dedupe_on( column_model=dedupe_model, default_model=None) self.dialog_merge_purge.set_handler_sort_on(column_model=sort_model, default_model=None) self.dialog_merge_purge.configure(source_path=model.filePath, dest_path=file_base + "_merged" + ext) self.dialog_merge_purge.show()
def test_remove_columns_random(self, dataFrame): columnNames = dataFrame.columns.tolist() columnNames = [(i, n) for i, n in enumerate(columnNames)] for cycle in range(1000): elements = random.randint(1, len(columnNames)) names = random.sample(columnNames, elements) df = dataFrame.copy() model = DataFrameModel(df) assert not model.removeDataFrameColumns(names) model.enableEditing(True) model.removeDataFrameColumns(names) _columnSet = set(columnNames) _removedSet = set(names) remainingColumns = _columnSet - _removedSet for idx, col in remainingColumns: assert col in model.dataFrame().columns.tolist()
def test_initDataFrameWithDataFrame(): dataFrame = pandas.DataFrame([0], columns=['A']) model = DataFrameModel(dataFrame) assert not model.dataFrame().empty assert model.dataFrame() is dataFrame
def test_initDataFrame(): model = DataFrameModel() assert model.dataFrame().empty