def save_graph(self):
from Orange.widgets.data.owsave import OWSave
save_img = OWSave(data={'scene': self.scene, 'tree': self.tree},
file_formats=dict(chain(
FileFormat.img_writers.items(),
FileFormat.graph_writers.items())))
save_img.exec_()
def test_extension_from_filter(self):
self.assertEqual(
OWSave._extension_from_filter("Description (*.ext)"), ".ext")
self.assertEqual(
OWSave._extension_from_filter("Description (*.foo.ba)"), ".foo.ba")
self.assertEqual(
OWSave._extension_from_filter("Description (.ext)"), ".ext")
self.assertEqual(
OWSave._extension_from_filter("Description (.foo.bar)"), ".foo.bar")
def test_extension_from_filter(self):
self.assertEqual(
OWSave._extension_from_filter("Description (*.ext)"), ".ext")
self.assertEqual(
OWSave._extension_from_filter("Description (*.foo.ba)"), ".foo.ba")
self.assertEqual(
OWSave._extension_from_filter("Description (.ext)"), ".ext")
self.assertEqual(
OWSave._extension_from_filter("Description (.foo.bar)"), ".foo.bar")
def save_graph(self):
from Orange.widgets.data.owsave import OWSave
save_img = OWSave(data={
'scene': self.scene,
'tree': self.tree
},
file_formats=dict(
chain(FileFormat.img_writers.items(),
FileFormat.graph_writers.items())))
save_img.exec_()
def test_save_file_dialog_enforces_extension_linux(self):
dialog = OWSave.SaveFileDialog(
OWSave, None, "Save File", "foo.bar",
"Bar files (*.tab);;Low files (*.csv)")
dialog.selectNameFilter("Low files (*.csv)")
self.assertTrue(dialog.selectedFiles()[0].endswith("/foo.csv"))
dialog.selectFile("high.bar")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.bar.csv"))
dialog.selectNameFilter("Bar files (*.tab)")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.bar.tab"))
dialog.selectFile("middle.pkl")
self.assertTrue(dialog.selectedFiles()[0].endswith("/middle.tab"))
dialog.filterSelected.emit("Low files (*.csv)")
self.assertTrue(dialog.selectedFiles()[0].endswith("/middle.csv"))
dialog.selectFile("high.tab.gz")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.csv"))
dialog.selectFile("high.tab.gz.tab.tab.gz")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.csv"))
def test_save_file_dialog_asks_for_overwrite_darwin(self, msgbox, dlg):
def selected_files():
nonlocal attempts
attempts += 1
return [["old.tab", "new.tab"][attempts]]
widget = self.widget
widget._initial_start_dir = lambda: "baz"
for filter1 in widget.filters:
if OWSave._extension_from_filter(filter1) == ".tab":
break
widget.filter = filter1
instance = dlg.return_value
instance.exec.return_value = QFileDialog.Accepted
instance.selectedFiles = selected_files
instance.selectedNameFilter.return_value = self.remove_star(filter1)
attempts = -1
msgbox.question.return_value = msgbox.Yes = 1
self.assertEqual(widget.get_save_filename()[0], "old.tab")
attempts = -1
msgbox.question.return_value = msgbox.No = 0
self.assertEqual(widget.get_save_filename()[0], "new.tab")
def test_save_file_dialog_asks_for_overwrite_darwin(self, msgbox, dlg):
def selected_files():
nonlocal attempts
attempts += 1
return [["old.tab", "new.tab"][attempts]]
widget = self.widget
widget._initial_start_dir = lambda: "baz"
for filter1 in widget.filters:
if OWSave._extension_from_filter(filter1) == ".tab":
break
widget.filter = filter1
instance = dlg.return_value
instance.exec.return_value = QFileDialog.Accepted
instance.selectedFiles = selected_files
instance.selectedNameFilter.return_value = self.remove_star(filter1)
attempts = -1
msgbox.question.return_value = msgbox.Yes = 1
self.assertEqual(widget.get_save_filename()[0], "old.tab")
attempts = -1
msgbox.question.return_value = msgbox.No = 0
self.assertEqual(widget.get_save_filename()[0], "new.tab")
def test_filename_with_fix_extension(self):
def mock_fix_choice(ret):
f = lambda *x: ret
f.__dict__.update(fix_extension.__dict__)
return f
change_filter = iter([PickleReader, TabReader])
for file_choice, fix in [
[lambda *x: ("o.pickle", format_filter(TabReader)),
mock_fix_choice(fix_extension.CHANGE_EXT)],
[lambda *x: ("o.tab", format_filter(PickleReader)),
mock_fix_choice(fix_extension.CHANGE_FORMAT)],
[lambda *x: ("o.tab", format_filter(next(change_filter))),
mock_fix_choice(fix_extension.CANCEL)]
]:
with patch("AnyQt.QtWidgets.QFileDialog.getSaveFileName", file_choice),\
patch("Orange.widgets.utils.filedialogs.fix_extension", fix):
saved_filename, format, filter = \
open_filename_dialog_save(".", None, OWSave.get_writers(False))
self.assertEqual(saved_filename, "o.tab")
self.assertEqual(format, TabReader)
self.assertEqual(filter, format_filter(TabReader))
class TestOWSaveLinuxDialog(OWSaveTestBase):
def test_get_save_filename_linux(self):
widget = self.widget
widget.initial_start_dir = lambda: "baz"
widget.get_filters = lambda: dict.fromkeys("abc")
widget.filter = "b"
dlg = widget.SaveFileDialog = Mock() # pylint: disable=invalid-name
instance = dlg.return_value
instance.selectedFiles.return_value = ["foo"]
instance.selectedNameFilter.return_value = "bar"
self.assertEqual(widget.get_save_filename(), ("foo", "bar"))
self.assertEqual(dlg.call_args[0][3], "baz")
self.assertEqual(dlg.call_args[0][4], "a;;b;;c")
instance.selectNameFilter.assert_called_with("b")
instance.exec.return_value = QFileDialog.Rejected
self.assertEqual(widget.get_save_filename(), ("", ""))
@patch.object(OWSaveBase, "filters", OWSave.get_filters())
def test_save_file_dialog_enforces_extension_linux(self):
dialog = OWSave.SaveFileDialog(
OWSave, None, "Save File", "foo.bar",
"Bar files (*.tab);;Low files (*.csv)")
dialog.selectNameFilter("Low files (*.csv)")
self.assertTrue(dialog.selectedFiles()[0].endswith("/foo.csv"))
dialog.selectFile("high.bar")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.bar.csv"))
dialog.selectNameFilter("Bar files (*.tab)")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.bar.tab"))
dialog.selectFile("middle.pkl")
self.assertTrue(dialog.selectedFiles()[0].endswith("/middle.tab"))
dialog.filterSelected.emit("Low files (*.csv)")
self.assertTrue(dialog.selectedFiles()[0].endswith("/middle.csv"))
dialog.selectFile("high.tab.gz")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.csv"))
dialog.selectFile("high.tab.gz.tab.tab.gz")
self.assertTrue(dialog.selectedFiles()[0].endswith("/high.csv"))
def test_save_file_dialog_uses_valid_filters_linux(self):
widget = self.widget
widget.valid_filters = lambda: ["a (*.a)", "b (*.b)"]
widget.default_valid_filter = lambda: "a (*.a)"
dlg = widget.SaveFileDialog = Mock() # pylint: disable=invalid-name
instance = dlg.return_value
instance.exec.return_value = dlg.Rejected = QFileDialog.Rejected
widget.get_save_filename()
self.assertEqual(dlg.call_args[0][4], "a (*.a);;b (*.b)")
instance.selectNameFilter.assert_called_with("a (*.a)")
def test_save_file_dialog_enforces_extension_darwin(self, dlg):
widget = self.widget
for filter1 in widget.filters:
if OWSave._extension_from_filter(filter1) == ".tab":
break
for filter2 in widget.filters:
if OWSave._extension_from_filter(filter2) == ".csv.gz":
break
widget.filter = filter1
instance = dlg.return_value
instance.exec.return_value = QFileDialog.Accepted
instance.selectedNameFilter.return_value = self.remove_star(filter1)
instance.selectedFiles.return_value = ["foo"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.pkl"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.tab.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.csv.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.bar"]
self.assertEqual(widget.get_save_filename()[0], "foo.bar.tab")
instance.selectedNameFilter.return_value = self.remove_star(filter2)
instance.selectedFiles.return_value = ["foo"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.pkl"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.tab.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.csv.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.bar"]
self.assertEqual(widget.get_save_filename()[0], "foo.bar.csv.gz")
def test_save_file_dialog_enforces_extension_darwin(self, dlg):
widget = self.widget
for filter1 in widget.filters:
if OWSave._extension_from_filter(filter1) == ".tab":
break
for filter2 in widget.filters:
if OWSave._extension_from_filter(filter2) == ".csv.gz":
break
widget.filter = filter1
instance = dlg.return_value
instance.exec.return_value = QFileDialog.Accepted
instance.selectedNameFilter.return_value = self.remove_star(filter1)
instance.selectedFiles.return_value = ["foo"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.pkl"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.tab.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.csv.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.tab")
instance.selectedFiles.return_value = ["foo.bar"]
self.assertEqual(widget.get_save_filename()[0], "foo.bar.tab")
instance.selectedNameFilter.return_value = self.remove_star(filter2)
instance.selectedFiles.return_value = ["foo"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.pkl"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.tab.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.csv.gz"]
self.assertEqual(widget.get_save_filename()[0], "foo.csv.gz")
instance.selectedFiles.return_value = ["foo.bar"]
self.assertEqual(widget.get_save_filename()[0], "foo.bar.csv.gz")
def test_migration_to_version_2(self):
const_settings = {
'add_type_annotations': True, 'auto_save': False,
'controlAreaVisible': True, 'last_dir': '/home/joe/Desktop',
'__version__': 1}
# No compression, Tab-separated values
settings = {**const_settings,
'compress': False, 'compression': 'gzip (.gz)',
'filetype': 'Tab-separated values (.tab)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Tab-separated values (*.tab)"})
# Compression; ignore compression format (.xz is no longer supported)
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)',
'filetype': 'Tab-separated values (.tab)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Compressed Tab-separated values (*.tab.gz)"})
# No compression, Excel
settings = {**const_settings,
'compress': False, 'compression': 'lzma (.xz)',
'filetype': 'Microsoft Excel spreadsheet (.xlsx)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Microsoft Excel spreadsheet (*.xlsx)"})
# Excel with compression - compression must be ignored
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)',
'filetype': 'Microsoft Excel spreadsheet (.xlsx)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Microsoft Excel spreadsheet (*.xlsx)"})
# Missing filetype (is this possible?)
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)'}
OWSave.migrate_settings(settings)
self.assertTrue(settings["filter"] in OWSave.get_filters())
# Unsupported file format (is this possible?)
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)',
'filetype': 'Bar file (.bar)'}
OWSave.migrate_settings(settings)
self.assertTrue(settings["filter"] in OWSave.get_filters())
def saveScene(self):
from Orange.widgets.data.owsave import OWSave
save_img = OWSave(data=self.scene, file_formats=FileFormat.img_writers)
save_img.exec_()
def saveScene(self):
from Orange.widgets.data.owsave import OWSave
save_img = OWSave(parent=self, data=self.scene,
file_formats=FileFormats.img_writers)
save_img.exec_()
def save_graph(self):
from Orange.widgets.data.owsave import OWSave
save_img = OWSave(parent=self, data=self.plot.plotItem,
file_formats=FileFormats.img_writers)
save_img.exec_()
def save_graph(self):
from Orange.widgets.data.owsave import OWSave
save_img = OWSave(data=self.grid_widget,
file_formats=FileFormat.img_writers)
save_img.exec_()
def test_migration_to_version_2(self):
const_settings = {
'add_type_annotations': True, 'auto_save': False,
'controlAreaVisible': True, 'last_dir': '/home/joe/Desktop',
'__version__': 1}
# No compression, Tab-separated values
settings = {**const_settings,
'compress': False, 'compression': 'gzip (.gz)',
'filetype': 'Tab-separated values (.tab)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Tab-separated values (*.tab)"})
# Compression; ignore compression format (.xz is no longer supported)
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)',
'filetype': 'Tab-separated values (.tab)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Compressed Tab-separated values (*.tab.gz)"})
# No compression, Excel
settings = {**const_settings,
'compress': False, 'compression': 'lzma (.xz)',
'filetype': 'Microsoft Excel spreadsheet (.xlsx)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Microsoft Excel spreadsheet (*.xlsx)"})
# Excel with compression - compression must be ignored
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)',
'filetype': 'Microsoft Excel spreadsheet (.xlsx)'}
OWSave.migrate_settings(settings)
self.assertEqual(
settings,
{**const_settings,
"filter": "Microsoft Excel spreadsheet (*.xlsx)"})
# Missing filetype (is this possible?)
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)'}
OWSave.migrate_settings(settings)
self.assertTrue(settings["filter"] in OWSave.filters)
# Unsupported file format (is this possible?)
settings = {**const_settings,
'compress': True, 'compression': 'lzma (.xz)',
'filetype': 'Bar file (.bar)'}
OWSave.migrate_settings(settings)
self.assertTrue(settings["filter"] in OWSave.filters)
