@@ -13,15 +13,15 @@ def assertMatrixLabels(self, m, l, mt=None, lt=None):
13
13
assert np .all (
14
14
np .equal (
15
15
m .eval (),
16
- mt if mt is not None else self .data .as_matrix ()
16
+ mt if mt is not None else self .data .values
17
17
)
18
18
)
19
19
assert l == list (lt or self .data .columns )
20
20
21
21
def test_numpy_init (self ):
22
- m , l = utils .any_to_tensor_and_labels (self .data .as_matrix () )
22
+ m , l = utils .any_to_tensor_and_labels (self .data .values )
23
23
self .assertMatrixLabels (m , l , lt = ['x0' , 'x1' ])
24
- m , l = utils .any_to_tensor_and_labels (self .data .as_matrix () , labels = ['x2' , 'x3' ])
24
+ m , l = utils .any_to_tensor_and_labels (self .data .values , labels = ['x2' , 'x3' ])
25
25
self .assertMatrixLabels (m , l , lt = ['x2' , 'x3' ])
26
26
27
27
def test_pandas_init (self ):
@@ -32,42 +32,42 @@ def test_pandas_init(self):
32
32
33
33
def test_dict_input (self ):
34
34
m , l = utils .any_to_tensor_and_labels (self .data .to_dict ('dict' ))
35
- self .assertMatrixLabels (m , l , mt = self .data . as_matrix ( l ) , lt = l )
35
+ self .assertMatrixLabels (m , l , mt = self .data [ l ]. values , lt = l )
36
36
37
37
m , l = utils .any_to_tensor_and_labels (self .data .to_dict ('series' ))
38
- self .assertMatrixLabels (m , l , mt = self .data . as_matrix ( l ) , lt = l )
38
+ self .assertMatrixLabels (m , l , mt = self .data [ l ]. values , lt = l )
39
39
40
40
m , l = utils .any_to_tensor_and_labels (self .data .to_dict ('list' ))
41
- self .assertMatrixLabels (m , l , mt = self .data . as_matrix ( l ) , lt = l )
41
+ self .assertMatrixLabels (m , l , mt = self .data [ l ]. values , lt = l )
42
42
43
43
inp = {k : tt .as_tensor_variable (v ) for k , v in self .data .to_dict ('series' ).items ()}
44
44
m , l = utils .any_to_tensor_and_labels (inp )
45
- self .assertMatrixLabels (m , l , mt = self .data . as_matrix ( l ) , lt = l )
45
+ self .assertMatrixLabels (m , l , mt = self .data [ l ]. values , lt = l )
46
46
47
47
def test_list_input (self ):
48
- m , l = utils .any_to_tensor_and_labels (self .data .as_matrix () .tolist ())
48
+ m , l = utils .any_to_tensor_and_labels (self .data .values .tolist ())
49
49
self .assertMatrixLabels (m , l , lt = ['x0' , 'x1' ])
50
- m , l = utils .any_to_tensor_and_labels (self .data .as_matrix () .tolist (), labels = ['x2' , 'x3' ])
50
+ m , l = utils .any_to_tensor_and_labels (self .data .values .tolist (), labels = ['x2' , 'x3' ])
51
51
self .assertMatrixLabels (m , l , lt = ['x2' , 'x3' ])
52
52
53
53
def test_tensor_input (self ):
54
54
m , l = utils .any_to_tensor_and_labels (
55
- tt .as_tensor_variable (self .data .as_matrix () .tolist ()),
55
+ tt .as_tensor_variable (self .data .values .tolist ()),
56
56
labels = ['x0' , 'x1' ]
57
57
)
58
58
self .assertMatrixLabels (m , l , lt = ['x0' , 'x1' ])
59
59
m , l = utils .any_to_tensor_and_labels (
60
- tt .as_tensor_variable (self .data .as_matrix () .tolist ()),
60
+ tt .as_tensor_variable (self .data .values .tolist ()),
61
61
labels = ['x2' , 'x3' ])
62
62
self .assertMatrixLabels (m , l , lt = ['x2' , 'x3' ])
63
63
64
64
def test_user_mistakes (self ):
65
65
# no labels for tensor variable
66
66
with pytest .raises (
67
67
ValueError ):
68
- utils .any_to_tensor_and_labels (tt .as_tensor_variable (self .data .as_matrix () .tolist ()))
68
+ utils .any_to_tensor_and_labels (tt .as_tensor_variable (self .data .values .tolist ()))
69
69
# len of labels is bad
70
70
with pytest .raises (
71
71
ValueError ):
72
- utils .any_to_tensor_and_labels (self .data .as_matrix () .tolist (),
72
+ utils .any_to_tensor_and_labels (self .data .values .tolist (),
73
73
labels = ['x' ])
0 commit comments