def doubleClicked_callback(self, index: QModelIndex):
import pandas as pd
import numpy as np
import numpy.ma
if index.isValid() and index.column() == NodeCols.Value:
item = index.internalPointer()
obj = item.node['obj']['fullname']
args = str(item.node['args'])
data, _ = self.shell.get_obj_value(
'analyze_getval', obj, args)
if isinstance(data, (pd.DataFrame, pd.Index, pd.Series)):
dialog = DataFrameEditor(self)
dialog.setup_and_check(data)
elif isinstance(data, (np.ndarray, np.ma.MaskedArray)):
dialog = ArrayEditor(self)
dialog.setup_and_check(data, title='', readonly=True)
elif isinstance(data, (list, set, tuple, dict)):
dialog = CollectionsEditor(self)
dialog.setup(data, title='', readonly=True)
else:
return
dialog.show()
def test_dataframeeditor_edit_bool(qtbot, monkeypatch):
"""Test that bools are editible in df and false-y strs are detected."""
MockQMessageBox = Mock()
attr_to_patch = ('spyder.plugins.variableexplorer.widgets' +
'.dataframeeditor.QMessageBox')
monkeypatch.setattr(attr_to_patch, MockQMessageBox)
test_params = [numpy.bool_, numpy.bool, bool]
test_strs = ['foo', 'false', 'f', '0', '0.', '0.0', '', ' ']
expected_df = DataFrame([1, 0, 0, 0, 0, 0, 0, 0, 0], dtype=bool)
for bool_type in test_params:
test_df = DataFrame([0, 1, 1, 1, 1, 1, 1, 1, 0], dtype=bool_type)
dialog = DataFrameEditor()
assert dialog.setup_and_check(test_df, 'Test Dataframe')
dialog.show()
qtbot.waitForWindowShown(dialog)
view = dialog.dataTable
qtbot.keyClick(view, Qt.Key_Right)
for test_str in test_strs:
qtbot.keyClick(view, Qt.Key_Space)
qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace)
qtbot.keyClicks(view.focusWidget(), test_str)
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
assert not MockQMessageBox.critical.called
qtbot.wait(200)
dialog.accept()
qtbot.wait(500)
try:
assert (numpy.sum(expected_df[0].values == dialog.get_value().
values[:, 0]) == len(expected_df))
except AttributeError:
assert (numpy.sum(expected_df[0].as_matrix() == dialog.get_value().
as_matrix()[:, 0]) == len(expected_df))
def test_dataframeeditor_edit_overflow(qtbot, monkeypatch):
"""
Test that entry of an overflowing integer is caught and handled properly.
Integration regression test for issue #6114 .
"""
MockQMessageBox = Mock()
attr_to_patch = ('spyder.plugins.variableexplorer.widgets' +
'.dataframeeditor.QMessageBox')
monkeypatch.setattr(attr_to_patch, MockQMessageBox)
# Numpy doesn't raise the OverflowError for ints smaller than 64 bits
if not os.name == 'nt':
int32_bit_exponent = 66
else:
int32_bit_exponent = 34
test_parameters = [(1, numpy.int32, int32_bit_exponent),
(2, numpy.int64, 66)]
expected_df = DataFrame([5, 6, 7, 3, 4])
for idx, int_type, bit_exponet in test_parameters:
test_df = DataFrame(numpy.arange(0, 5), dtype=int_type)
dialog = DataFrameEditor()
assert dialog.setup_and_check(test_df, 'Test Dataframe')
dialog.show()
qtbot.waitForWindowShown(dialog)
view = dialog.dataTable
qtbot.keyClick(view, Qt.Key_Right)
qtbot.keyClicks(view, '5')
qtbot.keyClick(view, Qt.Key_Down)
qtbot.keyClick(view, Qt.Key_Space)
qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace)
qtbot.keyClicks(view.focusWidget(), str(int(2 ** bit_exponet)))
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
MockQMessageBox.critical.assert_called_with(ANY, "Error", ANY)
assert MockQMessageBox.critical.call_count == idx
qtbot.keyClicks(view, '7')
qtbot.keyClick(view, Qt.Key_Up)
qtbot.keyClicks(view, '6')
qtbot.keyClick(view, Qt.Key_Down)
qtbot.wait(200)
dialog.accept()
qtbot.wait(500)
try:
assert numpy.sum(expected_df[0].values ==
dialog.get_value().values) == len(expected_df)
except AttributeError:
assert numpy.sum(
expected_df[0].as_matrix() ==
dialog.get_value().as_matrix()) == len(expected_df)
def test_dataframeeditor_edit_overflow(qtbot, monkeypatch):
"""
Test that entry of an overflowing integer is caught and handled properly.
Integration regression test for spyder-ide/spyder#6114.
"""
MockQMessageBox = Mock()
attr_to_patch = ('spyder.plugins.variableexplorer.widgets' +
'.dataframeeditor.QMessageBox')
monkeypatch.setattr(attr_to_patch, MockQMessageBox)
# Numpy doesn't raise the OverflowError for ints smaller than 64 bits
if not os.name == 'nt':
int32_bit_exponent = 66
else:
int32_bit_exponent = 34
test_parameters = [(1, numpy.int32, int32_bit_exponent),
(2, numpy.int64, 66)]
expected_df = DataFrame([5, 6, 7, 3, 4])
for idx, int_type, bit_exponet in test_parameters:
test_df = DataFrame(numpy.arange(0, 5), dtype=int_type)
dialog = DataFrameEditor()
assert dialog.setup_and_check(test_df, 'Test Dataframe')
dialog.show()
qtbot.waitForWindowShown(dialog)
view = dialog.dataTable
qtbot.keyClick(view, Qt.Key_Right)
qtbot.keyClicks(view, '5')
qtbot.keyClick(view, Qt.Key_Down)
qtbot.keyClick(view, Qt.Key_Space)
qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace)
qtbot.keyClicks(view.focusWidget(), str(int(2 ** bit_exponet)))
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
MockQMessageBox.critical.assert_called_with(ANY, "Error", ANY)
assert MockQMessageBox.critical.call_count == idx
qtbot.keyClicks(view, '7')
qtbot.keyClick(view, Qt.Key_Up)
qtbot.keyClicks(view, '6')
qtbot.keyClick(view, Qt.Key_Down)
qtbot.wait(200)
dialog.accept()
qtbot.wait(500)
try:
assert numpy.sum(expected_df[0].values ==
dialog.get_value().values) == len(expected_df)
except AttributeError:
assert numpy.sum(
expected_df[0].as_matrix() ==
dialog.get_value().as_matrix()) == len(expected_df)
def test_dataframeeditor_edit_complex(qtbot, monkeypatch):
"""
Test that editing complex dtypes is handled gracefully in df editor.
Integration regression test for issue #6115 .
"""
MockQMessageBox = Mock()
attr_to_patch = ('spyder.plugins.variableexplorer.widgets' +
'.dataframeeditor.QMessageBox')
monkeypatch.setattr(attr_to_patch, MockQMessageBox)
test_params = [(1, numpy.complex128), (2, numpy.complex64), (3, complex)]
for count, complex_type in test_params:
test_df = DataFrame(numpy.arange(10, 15), dtype=complex_type)
dialog = DataFrameEditor()
assert dialog.setup_and_check(test_df, 'Test Dataframe')
dialog.show()
qtbot.waitForWindowShown(dialog)
view = dialog.dataTable
qtbot.keyClick(view, Qt.Key_Right)
qtbot.keyClick(view, Qt.Key_Down)
qtbot.keyClick(view, Qt.Key_Space)
qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace)
qtbot.keyClicks(view.focusWidget(), "42")
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
MockQMessageBox.critical.assert_called_with(ANY, "Error", ANY)
assert MockQMessageBox.critical.call_count == count * 2 - 1
qtbot.keyClick(view, Qt.Key_Down)
qtbot.keyClick(view, '1')
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
MockQMessageBox.critical.assert_called_with(
ANY, "Error", ("Editing dtype {0!s} not yet supported."
.format(type(test_df.iloc[1, 0]).__name__)))
assert MockQMessageBox.critical.call_count == count * 2
qtbot.wait(200)
dialog.accept()
qtbot.wait(500)
try:
assert numpy.sum(test_df[0].values ==
dialog.get_value().values) == len(test_df)
except AttributeError:
assert numpy.sum(test_df[0].as_matrix() ==
dialog.get_value().as_matrix()) == len(test_df)
def test_dataframeeditor_edit_complex(qtbot, monkeypatch):
"""
Test that editing complex dtypes is handled gracefully in df editor.
Integration regression test for spyder-ide/spyder#6115.
"""
MockQMessageBox = Mock()
attr_to_patch = ('spyder.plugins.variableexplorer.widgets' +
'.dataframeeditor.QMessageBox')
monkeypatch.setattr(attr_to_patch, MockQMessageBox)
test_params = [(1, numpy.complex128), (2, numpy.complex64), (3, complex)]
for count, complex_type in test_params:
test_df = DataFrame(numpy.arange(10, 15), dtype=complex_type)
dialog = DataFrameEditor()
assert dialog.setup_and_check(test_df, 'Test Dataframe')
dialog.show()
qtbot.waitForWindowShown(dialog)
view = dialog.dataTable
qtbot.keyClick(view, Qt.Key_Right)
qtbot.keyClick(view, Qt.Key_Down)
qtbot.keyClick(view, Qt.Key_Space)
qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace)
qtbot.keyClicks(view.focusWidget(), "42")
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
MockQMessageBox.critical.assert_called_with(ANY, "Error", ANY)
assert MockQMessageBox.critical.call_count == count * 2 - 1
qtbot.keyClick(view, Qt.Key_Down)
qtbot.keyClick(view, '1')
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
MockQMessageBox.critical.assert_called_with(
ANY, "Error", ("Editing dtype {0!s} not yet supported.".format(
type(test_df.iloc[1, 0]).__name__)))
assert MockQMessageBox.critical.call_count == count * 2
qtbot.wait(200)
dialog.accept()
qtbot.wait(500)
try:
assert numpy.sum(
test_df[0].values == dialog.get_value().values) == len(test_df)
except AttributeError:
assert numpy.sum(test_df[0].as_matrix() ==
dialog.get_value().as_matrix()) == len(test_df)
def test_dataframe_to_type(qtbot):
"""Regression test for spyder-ide/spyder#12296"""
# Setup editor
d = {'col1': [1, 2], 'col2': [3, 4]}
df = DataFrame(data=d)
editor = DataFrameEditor()
assert editor.setup_and_check(df, 'Test DataFrame To action')
editor.show()
qtbot.waitForWindowShown(editor)
# Check editor doesn't have changes to save and select an initial element
assert not editor.btn_save_and_close.isEnabled()
view = editor.dataTable
view.setCurrentIndex(view.model().index(0, 0))
# Show context menu and select option `To bool`
view.menu.show()
qtbot.keyPress(view.menu, Qt.Key_Down)
qtbot.keyPress(view.menu, Qt.Key_Down)
qtbot.keyPress(view.menu, Qt.Key_Return)
# Check that changes where made from the editor
assert editor.btn_save_and_close.isEnabled()
def test_dataframeeditor_edit_bool(qtbot, monkeypatch):
"""Test that bools are editible in df and false-y strs are detected."""
MockQMessageBox = Mock()
attr_to_patch = ('spyder.plugins.variableexplorer.widgets' +
'.dataframeeditor.QMessageBox')
monkeypatch.setattr(attr_to_patch, MockQMessageBox)
test_params = [numpy.bool_, numpy.bool, bool]
test_strs = ['foo', 'false', 'f', '0', '0.', '0.0', '', ' ']
expected_df = DataFrame([1, 0, 0, 0, 0, 0, 0, 0, 0], dtype=bool)
for bool_type in test_params:
test_df = DataFrame([0, 1, 1, 1, 1, 1, 1, 1, 0], dtype=bool_type)
dialog = DataFrameEditor()
assert dialog.setup_and_check(test_df, 'Test Dataframe')
dialog.show()
qtbot.waitForWindowShown(dialog)
view = dialog.dataTable
qtbot.keyClick(view, Qt.Key_Right)
for test_str in test_strs:
qtbot.keyClick(view, Qt.Key_Space)
qtbot.keyClick(view.focusWidget(), Qt.Key_Backspace)
qtbot.keyClicks(view.focusWidget(), test_str)
qtbot.keyClick(view.focusWidget(), Qt.Key_Down)
assert not MockQMessageBox.critical.called
qtbot.wait(200)
dialog.accept()
qtbot.wait(500)
try:
assert (numpy.sum(expected_df[0].values ==
dialog.get_value().values[:, 0]) ==
len(expected_df))
except AttributeError:
assert (numpy.sum(expected_df[0].as_matrix() ==
dialog.get_value().as_matrix()[:, 0]) ==
len(expected_df))