File tree Expand file tree Collapse file tree 2 files changed +26
-1
lines changed Expand file tree Collapse file tree 2 files changed +26
-1
lines changed Original file line number Diff line number Diff line change @@ -4221,7 +4221,7 @@ def _mpdist(
4221
4221
return MPdist
4222
4222
4223
4223
4224
- def process_isconstant (T , m , T_subseq_isconstant , T_subseq_isfinite = None ):
4224
+ def process_isconstant (T , m , T_subseq_isconstant = None , T_subseq_isfinite = None ):
4225
4225
"""
4226
4226
A convenience wrapper around the `rolling_isconstant` and
4227
4227
`fix_isconstant_isfinite_conflicts`.
Original file line number Diff line number Diff line change @@ -1753,3 +1753,28 @@ def test_process_isconstant_2d():
1753
1753
T_subseq_isconstant_comp = core .process_isconstant (T , m , T_subseq_isconstant )
1754
1754
1755
1755
npt .assert_array_equal (T_subseq_isconstant_ref , T_subseq_isconstant_comp )
1756
+
1757
+
1758
+ def test_process_isconstant_1d_default ():
1759
+ # test the default value of `T_subseq_isconstant` in `process_isconstant`
1760
+ n = 64
1761
+ m = 8
1762
+
1763
+ # case 1: without nan
1764
+ T = np .random .rand (n )
1765
+ T [:m ] = 0.5 # constant subsequence
1766
+
1767
+ T_subseq_isconstant_ref = naive .rolling_isconstant (T , m , a_subseq_isconstant = None )
1768
+ T_subseq_isconstant_comp = core .process_isconstant (T , m , T_subseq_isconstant = None )
1769
+
1770
+ npt .assert_array_equal (T_subseq_isconstant_ref , T_subseq_isconstant_comp )
1771
+
1772
+ # case 2: with nan
1773
+ T = np .random .rand (n )
1774
+ T [:m ] = 0.5 # constant subsequence
1775
+ T [- m :] = np .nan # non-finite subsequence
1776
+
1777
+ T_subseq_isconstant_ref = naive .rolling_isconstant (T , m , a_subseq_isconstant = None )
1778
+ T_subseq_isconstant_comp = core .process_isconstant (T , m , T_subseq_isconstant = None )
1779
+
1780
+ npt .assert_array_equal (T_subseq_isconstant_ref , T_subseq_isconstant_comp )
You can’t perform that action at this time.
0 commit comments