Skip to content

Commit 7362c81

Browse files
committed
OWFile: Add test for loading sparse data
1 parent c553c5f commit 7362c81

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

Orange/widgets/data/tests/test_owfile.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,12 @@
22
# pylint: disable=missing-docstring
33
from os import path, remove
44
from unittest.mock import Mock
5+
import pickle
6+
import tempfile
7+
58

69
import numpy as np
10+
import scipy.sparse as sp
711

812
from AnyQt.QtCore import QMimeData, QPoint, Qt, QUrl
913
from AnyQt.QtGui import QDragEnterEvent, QDropEvent
@@ -195,3 +199,19 @@ def test_check_datetime_disabled(self):
195199
for i in range(4):
196200
vartype_delegate.setEditorData(combo, idx(i))
197201
self.assertEqual(combo.count(), counts[i])
202+
203+
def test_domain_edit_on_sparse_data(self):
204+
iris = Table("iris")
205+
iris.X = sp.csr_matrix(iris.X)
206+
207+
f = tempfile.NamedTemporaryFile(suffix='.pickle', delete=False)
208+
pickle.dump(iris, f)
209+
f.close()
210+
211+
self.widget.add_path(f.name)
212+
self.widget.load_data()
213+
214+
output = self.get_output("Data")
215+
self.assertIsInstance(output, Table)
216+
self.assertEqual(iris.X.shape, output.X.shape)
217+
self.assertTrue(sp.issparse(output.X))

0 commit comments

Comments
 (0)