def test_automatic_column_width(qtbot):
browser = NamespaceBrowser(None)
browser.set_shellwidget(Mock())
browser.setup()
col_width = [browser.editor.columnWidth(i) for i in range(4)]
browser.set_data({
'a_variable': {
'type': 'int',
'size': 1,
'view': '1',
'python_type': 'int',
'numpy_type': 'Unknown'
}
})
new_col_width = [browser.editor.columnWidth(i) for i in range(4)]
assert browser.editor.automatic_column_width
assert col_width != new_col_width # Automatic col width is on
browser.editor.horizontalHeader()._handle_section_is_pressed = True
browser.editor.setColumnWidth(0, 100) # Simulate user changing col width
assert browser.editor.automatic_column_width == False
browser.set_data({
'a_lengthy_variable_name_which_should_change_width': {
'type': 'int',
'size': 1,
'view': '1',
'python_type': 'int',
'numpy_type': 'Unknown'
}
})
assert browser.editor.columnWidth(0) == 100 # Automatic col width is off
def test_automatic_column_width(qtbot):
browser = NamespaceBrowser(None)
browser.set_shellwidget(Mock())
browser.setup(exclude_private=True,
exclude_uppercase=True,
exclude_capitalized=True,
exclude_unsupported=True,
minmax=False)
col_width = [browser.editor.columnWidth(i) for i in range(4)]
browser.set_data({
'a_variable': {
'type': 'int',
'size': 1,
'color': '#0000ff',
'view': '1'
}
})
new_col_width = [browser.editor.columnWidth(i) for i in range(4)]
assert browser.editor.automatic_column_width
assert col_width != new_col_width # Automatic col width is on
browser.editor.horizontalHeader()._handle_section_is_pressed = True
browser.editor.setColumnWidth(0, 100) # Simulate user changing col width
assert browser.editor.automatic_column_width == False
browser.set_data({
'a_lengthy_variable_name_which_should_change_width': {
'type': 'int',
'size': 1,
'color': '#0000ff',
'view': '1'
}
})
assert browser.editor.columnWidth(0) == 100 # Automatic col width is off
def test_setup_sets_dataframe_format(qtbot):
browser = NamespaceBrowser(None)
browser.set_shellwidget(Mock())
browser.setup(exclude_private=True, exclude_uppercase=True,
exclude_capitalized=True, exclude_unsupported=True,
minmax=False, dataframe_format='%10.5f')
assert browser.editor.model.dataframe_format == '%10.5f'
def namespacebrowser(qtbot):
browser = NamespaceBrowser(None)
browser.set_shellwidget(Mock())
browser.setup()
browser.resize(640, 480)
browser.show()
qtbot.addWidget(browser)
return browser
def test_sort_by_column(qtbot):
"""
Test that clicking the header view the namespacebrowser is sorted.
Regression test for spyder-ide/spyder#9835 .
"""
browser = NamespaceBrowser(None)
qtbot.addWidget(browser)
browser.set_shellwidget(Mock())
browser.setup(exclude_private=True,
exclude_uppercase=True,
exclude_capitalized=True,
exclude_unsupported=False,
exclude_callables_and_modules=True,
minmax=False)
browser.set_data({
'a_variable': {
'type': 'int',
'size': 1,
'color': '#0000ff',
'view': '1'
},
'b_variable': {
'type': 'int',
'size': 1,
'color': '#0000ff',
'view': '2'
}
})
header = browser.editor.horizontalHeader()
# Check header is clickable
assert header.sectionsClickable()
model = browser.editor.model
# Base check of the model
assert model.rowCount() == 2
assert model.columnCount() == 5
assert data_table(model, 2, 4) == [['a_variable', 'b_variable'],
['int', 'int'], [1, 1], ['1', '2']]
with qtbot.waitSignal(header.sectionClicked):
browser.show()
qtbot.mouseClick(header.viewport(), Qt.LeftButton, pos=QPoint(1, 1))
# Check sort effect
assert data_table(model, 2, 4) == [['b_variable', 'a_variable'],
['int', 'int'], [1, 1], ['2', '1']]
def test_sort_by_column(qtbot):
"""
Test that clicking the header view the namespacebrowser is sorted.
Regression test for spyder-ide/spyder#9835 .
"""
browser = NamespaceBrowser(None)
qtbot.addWidget(browser)
browser.set_shellwidget(Mock())
browser.setup()
browser.set_data({
'a_variable': {
'type': 'int',
'size': 1,
'view': '1',
'python_type': 'int',
'numpy_type': 'Unknown'
},
'b_variable': {
'type': 'int',
'size': 1,
'view': '2',
'python_type': 'int',
'numpy_type': 'Unknown'
}
})
header = browser.editor.horizontalHeader()
# Check header is clickable
assert header.sectionsClickable()
model = browser.editor.model
# Base check of the model
assert model.rowCount() == 2
assert model.columnCount() == 5
assert data_table(model, 2, 4) == [['a_variable', 'b_variable'],
['int', 'int'], [1, 1], ['1', '2']]
with qtbot.waitSignal(header.sectionClicked):
browser.show()
qtbot.mouseClick(header.viewport(), Qt.LeftButton, pos=QPoint(1, 1))
# Check sort effect
assert data_table(model, 2, 4) == [['b_variable', 'a_variable'],
['int', 'int'], [1, 1], ['2', '1']]
def test_keys_sorted_and_sort_with_large_rows(qtbot):
"""
Test that keys are sorted and sorting works as expected when
there's a large number of rows.
This is a regression test for issue spyder-ide/spyder#10702
"""
browser = NamespaceBrowser(None)
qtbot.addWidget(browser)
browser.set_shellwidget(Mock())
browser.setup()
# Create variables.
variables = {}
variables['i'] = ({
'type': 'int',
'size': 1,
'view': '1',
'python_type': 'int',
'numpy_type': 'Unknown'
})
for i in range(100):
if i < 10:
var = 'd_0' + str(i)
else:
var = 'd_' + str(i)
variables[var] = ({
'type': 'int',
'size': 1,
'view': '1',
'python_type': 'int',
'numpy_type': 'Unknown'
})
# Set data
browser.set_data(variables)
# Assert we loaded the expected amount of data and that we can fetch
# more.
model = browser.editor.model
assert model.rowCount() == ROWS_TO_LOAD
assert model.canFetchMore(QModelIndex())
# Assert keys are sorted
assert data(model, 49, 0) == 'd_49'
# Sort
header = browser.editor.horizontalHeader()
with qtbot.waitSignal(header.sectionClicked):
qtbot.mouseClick(header.viewport(), Qt.LeftButton, pos=QPoint(1, 1))
# Assert we loaded all data before performing the sort.
assert data(model, 0, 0) == 'i'
def test_setup_sets_dataframe_format(qtbot):
browser = NamespaceBrowser(None)
browser.set_shellwidget(Mock())
browser.setup(exclude_private=True, exclude_uppercase=True,
exclude_capitalized=True, exclude_unsupported=True,
minmax=False, dataframe_format='%10.5f')
assert browser.editor.source_model.dataframe_format == '%10.5f'
def test_automatic_column_width(qtbot):
browser = NamespaceBrowser(None)
browser.set_shellwidget(Mock())
browser.setup(exclude_private=True, exclude_uppercase=True,
exclude_capitalized=True, exclude_unsupported=True,
minmax=False)
col_width = [browser.editor.columnWidth(i) for i in range(4)]
browser.set_data({'a_variable':
{'type': 'int', 'size': 1, 'color': '#0000ff', 'view': '1'}})
new_col_width = [browser.editor.columnWidth(i) for i in range(4)]
assert browser.editor.automatic_column_width
assert col_width != new_col_width # Automatic col width is on
browser.editor.horizontalHeader()._handle_section_is_pressed = True
browser.editor.setColumnWidth(0, 100) # Simulate user changing col width
assert browser.editor.automatic_column_width == False
browser.set_data({'a_lengthy_variable_name_which_should_change_width':
{'type': 'int', 'size': 1, 'color': '#0000ff', 'view': '1'}})
assert browser.editor.columnWidth(0) == 100 # Automatic col width is off
def create_new_widget(self, shellwidget):
nsb = NamespaceBrowser(self)
nsb.set_shellwidget(shellwidget)
nsb.setup()
nsb.sig_free_memory_requested.connect(self.free_memory)
nsb.sig_start_spinner_requested.connect(self.start_spinner)
nsb.sig_stop_spinner_requested.connect(self.stop_spinner)
nsb.sig_hide_finder_requested.connect(self.hide_finder)
self._set_actions_and_menus(nsb)
return nsb
def add_shellwidget(self, shellwidget):
"""
Register shell with variable explorer.
This function creates a new NamespaceBrowser for browsing
variables in the shell.
"""
shellwidget_id = id(shellwidget)
if shellwidget_id not in self._shellwidgets:
nsb = NamespaceBrowser(self)
nsb.set_shellwidget(shellwidget)
nsb.setup()
self.add_widget(nsb)
self._set_actions_and_menus(nsb)
self._shellwidgets[shellwidget_id] = nsb
self.set_current_widget(nsb)
self.update_actions()
return nsb
def add_shellwidget(self, shellwidget):
"""
Register shell with variable explorer.
This function opens a new NamespaceBrowser for browsing the variables
in the shell.
"""
shellwidget_id = id(shellwidget)
if shellwidget_id not in self.shellwidgets:
self.options_button.setVisible(True)
nsb = NamespaceBrowser(self, options_button=self.options_button)
nsb.set_shellwidget(shellwidget)
nsb.setup(**self.get_settings())
nsb.sig_option_changed.connect(self.change_option)
nsb.sig_free_memory.connect(self.free_memory)
self.add_widget(nsb)
self.shellwidgets[shellwidget_id] = nsb
self.set_shellwidget_from_id(shellwidget_id)
return nsb
def test_filtering_with_large_rows(qtbot):
"""
Test that filtering works when there's a large number of rows.
"""
browser = NamespaceBrowser(None)
qtbot.addWidget(browser)
browser.set_shellwidget(Mock())
browser.setup()
text_finder = NamespacesBrowserFinder(browser.editor,
callback=browser.editor.set_regex,
main=browser,
regex_base=VALID_VARIABLE_CHARS)
browser.set_text_finder(text_finder)
# Create data
variables = {}
for i in range(200):
letter = string.ascii_lowercase[i // 10]
var = letter + str(i)
variables[var] = ({
'type': 'int',
'size': 1,
'view': '1',
'python_type': 'int',
'numpy_type': 'Unknown'
})
# Set data
browser.set_data(variables)
# Assert we loaded the expected amount of data and that we can fetch
# more data.
model = browser.editor.model
assert model.rowCount() == ROWS_TO_LOAD
assert model.canFetchMore(QModelIndex())
assert data(model, 49, 0) == 'e49'
# Assert we can filter variables not loaded yet.
qtbot.keyClicks(text_finder, "t19")
assert model.rowCount() == 10
# Assert all variables effectively start with 't19'.
for i in range(10):
assert data(model, i, 0) == 't19{}'.format(i)
# Reset text_finder widget.
text_finder.setText('')
# Create a new variable that starts with a different letter than
# the rest.
new_variables = variables.copy()
new_variables['z'] = ({
'type': 'int',
'size': 1,
'view': '1',
'python_type': 'int',
'numpy_type': 'Unknown'
})
# Emulate the process of loading those variables after the
# namespace view is sent from the kernel.
browser.process_remote_view(new_variables)
# Assert that can find 'z' among the declared variables.
qtbot.keyClicks(text_finder, "z")
assert model.rowCount() == 1
def test_filtering_with_large_rows(qtbot):
"""
Test that filtering works when there's a large number of rows.
"""
browser = NamespaceBrowser(None)
qtbot.addWidget(browser)
browser.set_shellwidget(Mock())
browser.setup(exclude_private=True, exclude_uppercase=True,
exclude_capitalized=True, exclude_unsupported=False,
exclude_callables_and_modules=True,
minmax=False)
# Create data
variables = {}
for i in range(200):
letter = string.ascii_lowercase[i // 10]
var = letter + str(i)
variables[var] = (
{'type': 'int', 'size': 1, 'color': '#0000ff', 'view': '1'}
)
# Set data
browser.set_data(variables)
# Assert we loaded the expected amount of data and that we can fetch
# more data.
model = browser.editor.model
assert model.rowCount() == ROWS_TO_LOAD
assert model.canFetchMore(QModelIndex())
assert data(model, 49, 0) == 'e49'
# Assert we can filter variables not loaded yet.
qtbot.keyClicks(browser.finder.text_finder, "t19")
assert model.rowCount() == 10
# Assert all variables effectively start with 't19'.
for i in range(10):
assert data(model, i, 0) == 't19{}'.format(i)
# Hide finder widget in order to reset it.
browser.show_finder(set_visible=False)
# Create a new variable that starts with a different letter than
# the rest.
new_variables = variables.copy()
new_variables['z'] = (
{'type': 'int', 'size': 1, 'color': '#0000ff', 'view': '1'}
)
# Emulate the process of loading those variables after the
# namespace view is sent from the kernel.
browser.process_remote_view(new_variables)
# Assert that can find 'z' among the declared variables.
qtbot.keyClicks(browser.finder.text_finder, "z")
assert model.rowCount() == 1