1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
# Test methods with long descriptive names can omit docstrings
# pylint: disable=missing-docstring
import unittest
from Orange.data import Table
from Orange.preprocess import ProjectCUR
from Orange.tests import test_filename
class TestCURProjector(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.ionosphere = Table(test_filename("datasets/ionosphere.tab"))
def test_project_cur_default(self):
data = self.ionosphere
projector = ProjectCUR()
data_cur = projector(data)
for i in range(data_cur.X.shape[1]):
sbtr = (data.X - data_cur.X[:, i][:, None]) == 0
self.assertTrue(((sbtr.sum(0) == data.X.shape[0])).any())
self.assertTrue(data_cur.X.shape[1] <= data.X.shape[1])
self.assertTrue((data.metas == data_cur.metas).all())
self.assertTrue((data.Y == data_cur.Y).any())
def test_project_cur(self):
data = self.ionosphere
projector = ProjectCUR(rank=3, max_error=1)
data_cur = projector(data)
for i in range(data_cur.X.shape[1]):
sbtr = (data.X - data_cur.X[:, i][:, None]) == 0
self.assertTrue(((sbtr.sum(0) == data.X.shape[0])).any())
self.assertTrue(data_cur.X.shape[1] <= data.X.shape[1])
self.assertTrue((data.metas == data_cur.metas).all())
self.assertTrue((data.Y == data_cur.Y).any())
|
