File: test_sparse_table.py

package info (click to toggle)
orange3 3.40.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 15,908 kB
  • sloc: python: 162,745; ansic: 622; makefile: 322; sh: 93; cpp: 77
file content (67 lines) | stat: -rw-r--r-- 2,447 bytes parent folder | download | duplicates (2)
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# Test methods with long descriptive names can omit docstrings
# pylint: disable=missing-docstring
import warnings

import numpy as np
from scipy.sparse import csr_matrix, SparseEfficiencyWarning

from Orange import data
from Orange.data import Table
from Orange.tests import test_table as tabletests


class InterfaceTest(tabletests.InterfaceTest):
    def setUp(self):
        super().setUp()
        self.table = data.Table.from_numpy(
            self.domain,
            csr_matrix(self.table.X),
            csr_matrix(self.table.Y),
        )

    def test_row_assignment(self):
        # This warning will appear in actual code that assigns rows of
        # sparse matrix, which is OK, but We don't need it in test outputs
        warnings.filterwarnings("ignore", ".*", SparseEfficiencyWarning)
        super().test_row_assignment()

    def test_value_assignment(self):
        # This warning will appear in actual code that assigns rows of
        # sparse matrix, which is OK, but We don't need it in test outputs
        warnings.filterwarnings("ignore", ".*", SparseEfficiencyWarning)
        super().test_value_assignment()

    def test_str(self):
        iris = Table('iris')
        with iris.unlocked():
            iris.X, iris.Y = csr_matrix(iris.X), csr_matrix(iris.Y)
        str(iris)

    def test_Y_setter_1d(self):
        iris = Table('iris')
        assert iris.Y.shape == (150,)
        with iris.unlocked():
            iris.Y = csr_matrix(iris.Y)
        # We expect the Y shape to match the X shape, which is (150, 4) in iris
        self.assertEqual(iris.Y.shape, (150,))

    def test_Y_setter_2d(self):
        iris = Table('iris')
        assert iris.Y.shape == (150,)
        # Convert iris.Y to (150, 1) shape
        new_y = iris.Y[:, np.newaxis]
        with iris.unlocked():
            iris.Y = np.hstack((new_y, new_y))
            iris.Y = csr_matrix(iris.Y)
        # We expect the Y shape to match the X shape, which is (150, 4) in iris
        self.assertEqual(iris.Y.shape, (150, 2))

    def test_Y_setter_2d_single_instance(self):
        iris = Table('iris')[:1]
        # Convert iris.Y to (1, 1) shape
        new_y = iris.Y[:, np.newaxis]
        with iris.unlocked_reference():
            iris.Y = np.hstack((new_y, new_y))
            iris.Y = csr_matrix(iris.Y)
        # We expect the Y shape to match the X shape, which is (1, 4) in iris
        self.assertEqual(iris.Y.shape, (1, 2))