1
- """Tests for CommonGridBinner module."""
1
+ """Tests for GridBinner module."""
2
2
3
3
import re
4
4
from typing import Any , Tuple , Union
7
7
import pytest
8
8
from numpy .typing import NDArray
9
9
10
- from skfda .preprocessing .binning import CommonGridBinner
10
+ from skfda .preprocessing .binning import GridBinner
11
11
from skfda .representation import FData , FDataGrid , FDataIrregular
12
12
13
13
RaisesFixture = Tuple [
19
19
str ,
20
20
str ,
21
21
]
22
- ParamFixture = Tuple [FData , CommonGridBinner , Tuple [NDArray [np .float64 ], ...]]
22
+ ParamFixture = Tuple [FData , GridBinner , Tuple [NDArray [np .float64 ], ...]]
23
23
24
24
25
25
##############################################################################
82
82
],
83
83
],
84
84
)
85
- def commongridbinner_raises_fixture (request : Any ) -> RaisesFixture :
86
- """Fixture for getting a CommonGridBinner that raises a ValueError."""
85
+ def gridbinner_raises_fixture (request : Any ) -> RaisesFixture :
86
+ """Fixture for getting a GridBinner that raises a ValueError."""
87
87
return tuple (request .param )
88
88
89
89
90
90
@pytest .fixture (
91
91
params = [
92
92
[
93
- CommonGridBinner (bins = (1 , 1 )),
93
+ GridBinner (bins = (1 , 1 )),
94
94
fd .domain_range ,
95
95
],
96
96
[
97
- CommonGridBinner (bins = (1 , 1 ), domain_range = ((0 , 2 ), (0 , 2 ))),
97
+ GridBinner (bins = (1 , 1 ), domain_range = ((0 , 2 ), (0 , 2 ))),
98
98
((0 , 2 ), (0 , 2 )),
99
99
],
100
100
[
101
- CommonGridBinner (bins = (np .array ([- 1 , 1 ]), np .array ([- 1 , 1 ]))),
101
+ GridBinner (bins = (np .array ([- 1 , 1 ]), np .array ([- 1 , 1 ]))),
102
102
((- 1 , 1 ), (- 1 , 1 )),
103
103
],
104
104
[
105
- CommonGridBinner (
105
+ GridBinner (
106
106
bins = (np .array ([- 1 , 1 ]), np .array ([- 1 , 1 ])),
107
107
domain_range = ((0 , 2 ), (0 , 2 )),
108
108
),
@@ -120,22 +120,22 @@ def domain_range_interactions_fixture(
120
120
@pytest .fixture (
121
121
params = [
122
122
[
123
- CommonGridBinner (
123
+ GridBinner (
124
124
bins = (2 , 2 ),
125
125
output_grid = (np .array ([0 , 1 ]), np .array ([0 , 1 ])),
126
126
),
127
127
(np .array ([0 , 1 ]), np .array ([0 , 1 ])),
128
128
],
129
129
[
130
- CommonGridBinner (bins = (2 , 2 ), output_grid = "left" ),
130
+ GridBinner (bins = (2 , 2 ), output_grid = "left" ),
131
131
(np .array ([0 , 0.5 ]), np .array ([0 , 0.5 ])),
132
132
],
133
133
[
134
- CommonGridBinner (bins = (2 , 2 ), output_grid = "right" ),
134
+ GridBinner (bins = (2 , 2 ), output_grid = "right" ),
135
135
(np .array ([0.5 , 1 ]), np .array ([0.5 , 1 ])),
136
136
],
137
137
[
138
- CommonGridBinner (bins = (2 , 2 ), output_grid = "middle" ),
138
+ GridBinner (bins = (2 , 2 ), output_grid = "middle" ),
139
139
(np .array ([0.25 , 0.75 ]), np .array ([0.25 , 0.75 ])),
140
140
],
141
141
],
@@ -150,11 +150,11 @@ def output_grid_interactions_fixture(
150
150
@pytest .fixture (
151
151
params = [
152
152
[
153
- CommonGridBinner (bins = 2 ),
153
+ GridBinner (bins = 2 ),
154
154
"Input FData must have 1 domain dimensions." ,
155
155
],
156
156
[
157
- CommonGridBinner (
157
+ GridBinner (
158
158
bins = (2 , 2 ),
159
159
output_grid = (np .array ([0.6 , 1 ]), np .array ([0 , 1 ])),
160
160
),
@@ -166,7 +166,7 @@ def output_grid_interactions_fixture(
166
166
)
167
167
def fitting_raises_fixture (
168
168
request : Any ,
169
- ) -> Tuple [FDataGrid , CommonGridBinner , str ]:
169
+ ) -> Tuple [FDataGrid , GridBinner , str ]:
170
170
"""Fixture for checking mismatch between binner and FData."""
171
171
return fd , * request .param
172
172
@@ -444,17 +444,17 @@ def precalc_example_data_domain_3(
444
444
##############################################################################
445
445
446
446
447
- def test_raises (commongridbinner_raises_fixture : RaisesFixture ) -> None :
447
+ def test_raises (gridbinner_raises_fixture : RaisesFixture ) -> None :
448
448
"""
449
449
Check raises ValueError.
450
450
451
- Check that CommonGridBinners raise a ValueError exception.
451
+ Check that GridBinners raise a ValueError exception.
452
452
"""
453
- bins , domain_range_ , output_grid , * rest = commongridbinner_raises_fixture
453
+ bins , domain_range_ , output_grid , * rest = gridbinner_raises_fixture
454
454
bin_aggregation , error_msg = rest
455
455
456
456
with pytest .raises (ValueError , match = error_msg ):
457
- CommonGridBinner (
457
+ GridBinner (
458
458
bins = bins ,
459
459
domain_range = domain_range_ ,
460
460
output_grid = output_grid ,
@@ -499,7 +499,7 @@ def test_output_grid_interactions(
499
499
500
500
501
501
def test_fitting_raises (
502
- fitting_raises_fixture : Tuple [FDataGrid , CommonGridBinner , str ],
502
+ fitting_raises_fixture : Tuple [FDataGrid , GridBinner , str ],
503
503
) -> None :
504
504
"""
505
505
Check the domain parameter behaviour in fitting.
@@ -519,12 +519,12 @@ def test_precalc_example_domain_1(
519
519
Check the precalculated example for binned FData.
520
520
521
521
Compare the theoretical precalculated results against the binned data
522
- with CommonGridBinner implementation, for different FData.
522
+ with GridBinner implementation, for different FData.
523
523
524
524
Dimensions interacting are 1->n.
525
525
"""
526
526
fd , precalc_result = precalc_example_data_domain_1
527
- binner = CommonGridBinner (bins = 2 )
527
+ binner = GridBinner (bins = 2 )
528
528
computed_result = binner .fit_transform (fd )
529
529
530
530
np .testing .assert_array_equal (
@@ -545,12 +545,12 @@ def test_precalc_example_domain_3(
545
545
Check the precalculated example for binned FData.
546
546
547
547
Compare the theoretical precalculated results against the binned data
548
- with CommonGridBinner implementation, for different FData.
548
+ with GridBinner implementation, for different FData.
549
549
550
550
Dimensions interacting are 3->n.
551
551
"""
552
552
fd , precalc_result = precalc_example_data_domain_3
553
- binner = CommonGridBinner (bins = (2 , 2 , 2 ))
553
+ binner = GridBinner (bins = (2 , 2 , 2 ))
554
554
computed_result = binner .fit_transform (fd )
555
555
556
556
np .testing .assert_allclose (
0 commit comments