2
2
3
3
import numpy as np
4
4
from persim import PersistenceImager
5
- from persim . images_weights import *
6
- from persim . images_kernels import *
5
+ from persim import images_kernels
6
+ from persim import images_weights
7
7
8
8
# ----------------------------------------
9
9
# New PersistenceImager Tests
@@ -155,18 +155,18 @@ def test_mixed_pairs():
155
155
156
156
class TestWeightFunctions :
157
157
def test_zero_on_birthaxis (self ):
158
- persimgr = PersistenceImager (weight = linear_ramp , weight_params = {'low' :0.0 , 'high' :1.0 , 'start' :0.0 , 'end' :1.0 })
158
+ persimgr = PersistenceImager (weight = images_weights . linear_ramp , weight_params = {'low' :0.0 , 'high' :1.0 , 'start' :0.0 , 'end' :1.0 })
159
159
wf = persimgr .weight
160
160
wf_params = persimgr .weight_params
161
161
np .testing .assert_equal (wf (1 , 0 , ** wf_params ), 0 )
162
162
163
- persimgr = PersistenceImager (weight = persistence , weight_params = {'n' : 2 })
163
+ persimgr = PersistenceImager (weight = images_weights . persistence , weight_params = {'n' : 2 })
164
164
wf = persimgr .weight
165
165
wf_params = persimgr .weight_params
166
166
np .testing .assert_equal (wf (1 , 0 , ** wf_params ), 0 )
167
167
168
168
def test_linear_ramp (self ):
169
- persimgr = PersistenceImager (weight = linear_ramp , weight_params = {'low' :0.0 , 'high' :5.0 , 'start' :0.0 , 'end' :1.0 })
169
+ persimgr = PersistenceImager (weight = images_weights . linear_ramp , weight_params = {'low' :0.0 , 'high' :5.0 , 'start' :0.0 , 'end' :1.0 })
170
170
171
171
wf = persimgr .weight
172
172
wf_params = persimgr .weight_params
@@ -198,7 +198,7 @@ def test_linear_ramp(self):
198
198
np .testing .assert_equal (wf (1 , 2 , ** wf_params ), 2 )
199
199
200
200
def test_persistence (self ):
201
- persimgr = PersistenceImager (weight = persistence , weight_params = {'n' :1.0 })
201
+ persimgr = PersistenceImager (weight = images_weights . persistence , weight_params = {'n' :1.0 })
202
202
203
203
wf = persimgr .weight
204
204
wf_params = persimgr .weight_params
@@ -213,11 +213,11 @@ def test_persistence(self):
213
213
214
214
class TestKernelFunctions :
215
215
def test_gaussian (self ):
216
- kernel = gaussian
216
+ kernel = images_kernels . gaussian
217
217
kernel_params = {'mu' :[1 , 1 ], 'sigma' :np .array ([[1 ,0 ],[0 ,1 ]])}
218
218
np .testing .assert_almost_equal (kernel (np .array ([1 ]), np .array ([1 ]), ** kernel_params ), 1 / 4 , 8 )
219
219
220
- kernel = bvn_cdf
220
+ kernel = images_kernels . bvn_cdf
221
221
kernel_params = {'mu_x' :1.0 , 'mu_y' :1.0 , 'sigma_xx' :1.0 , 'sigma_yy' :1.0 , 'sigma_xy' :0.0 }
222
222
np .testing .assert_almost_equal (kernel (np .array ([1 ]), np .array ([1 ]), ** kernel_params ), 1 / 4 , 8 )
223
223
@@ -231,19 +231,19 @@ def test_gaussian(self):
231
231
np .testing .assert_equal (kernel (np .array ([1 ]), np .array ([1 ]), ** kernel_params ), .375 )
232
232
233
233
def test_norm_cdf (self ):
234
- np .testing .assert_equal (norm_cdf (0 ), .5 )
235
- np .testing .assert_almost_equal (norm_cdf (1 ), 0.8413447460685429 , 8 )
234
+ np .testing .assert_equal (images_kernels . norm_cdf (0 ), .5 )
235
+ np .testing .assert_almost_equal (images_kernels . norm_cdf (1 ), 0.8413447460685429 , 8 )
236
236
237
237
def test_uniform (self ):
238
- kernel = uniform
238
+ kernel = images_kernels . uniform
239
239
kernel_params = {'width' : 3 , 'height' : 1 }
240
240
241
241
np .testing .assert_equal (kernel (np .array ([- 1 ]), np .array ([- 1 ]), mu = (0 ,0 ), ** kernel_params ), 0 )
242
242
np .testing .assert_equal (kernel (np .array ([3 ]), np .array ([1 ]), mu = (0 ,0 ), ** kernel_params ), 1 )
243
243
np .testing .assert_equal (kernel (np .array ([5 ]), np .array ([5 ]), mu = (0 ,0 ), ** kernel_params ), 1 )
244
244
245
245
def test_sigma (self ):
246
- kernel = gaussian
246
+ kernel = images_kernels . gaussian
247
247
kernel_params1 = {'sigma' :np .array ([[1 ,0 ],[0 ,1 ]])}
248
248
kernel_params2 = {'sigma' : [[1 ,0 ],[0 ,1 ]]}
249
249
np .testing .assert_equal (kernel (np .array ([1 ]), np .array ([1 ]), ** kernel_params1 ), kernel (np .array ([1 ]), np .array ([1 ]), ** kernel_params2 ))
0 commit comments