Skip to content

Commit 3538703

Browse files
author
Francis Motta
committed
replaced import * statements in images.py and test_persim_update.py with from persim import <module>
1 parent 70cc1f8 commit 3538703

File tree

4 files changed

+40
-38
lines changed

4 files changed

+40
-38
lines changed

docs/notebooks/PersImage_update.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@
168168
},
169169
{
170170
"cell_type": "code",
171-
"execution_count": 14,
171+
"execution_count": 1,
172172
"metadata": {},
173173
"outputs": [],
174174
"source": [
@@ -178,7 +178,7 @@
178178
},
179179
{
180180
"cell_type": "code",
181-
"execution_count": 15,
181+
"execution_count": 2,
182182
"metadata": {},
183183
"outputs": [],
184184
"source": [

docs/notebooks/Persistence Images.ipynb

Lines changed: 19 additions & 17 deletions
Large diffs are not rendered by default.

persim/images.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212
import scipy.spatial as spatial
1313
import matplotlib.pyplot as plt
1414
from matplotlib.collections import LineCollection
15-
from persim.images_weights import *
16-
from persim.images_kernels import *
15+
from persim import images_kernels
16+
from persim import images_weights
1717

1818
from sklearn.base import TransformerMixin, BaseEstimator
1919

@@ -309,9 +309,9 @@ def __init__(self, birth_range=None, pers_range=None, pixel_size=None,
309309
if pixel_size is None:
310310
pixel_size = np.min([pers_range[1] - pers_range[0], birth_range[1] - birth_range[0]]) * 0.2
311311
if weight is None:
312-
weight = persistence
312+
weight = images_weights.persistence
313313
if kernel is None:
314-
kernel = gaussian
314+
kernel = images_kernels.gaussian
315315
if weight_params is None:
316316
weight_params = {'n': 1.0}
317317
if kernel_params is None:
@@ -604,7 +604,7 @@ def _transform(pers_dgm, skew=True, resolution=None, weight=None, weight_params=
604604
wts = weight(pers_dgm[:, 0], pers_dgm[:, 1], **weight_params)
605605

606606
# handle the special case of a standard, isotropic Gaussian kernel
607-
if kernel == gaussian:
607+
if kernel == images_kernels.gaussian:
608608
general_flag = False
609609
sigma = kernel_params['sigma']
610610

@@ -615,8 +615,8 @@ def _transform(pers_dgm, skew=True, resolution=None, weight=None, weight_params=
615615
if (sigma[0][0] == sigma[1][1] and sigma[0][1] == 0.0):
616616
sigma = np.sqrt(sigma[0][0])
617617
for i in range(n):
618-
ncdf_b = norm_cdf((_bpnts - pers_dgm[i, 0]) / sigma)
619-
ncdf_p = norm_cdf((_ppnts - pers_dgm[i, 1]) / sigma)
618+
ncdf_b = images_kernels.norm_cdf((_bpnts - pers_dgm[i, 0]) / sigma)
619+
ncdf_p = images_kernels.norm_cdf((_ppnts - pers_dgm[i, 1]) / sigma)
620620
curr_img = ncdf_p[None, :] * ncdf_b[:, None]
621621
pers_img += wts[i]*(curr_img[1:, 1:] - curr_img[:-1, 1:] - curr_img[1:, :-1] + curr_img[:-1, :-1])
622622
else:

test/test_persim_update.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@
22

33
import numpy as np
44
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
77

88
# ----------------------------------------
99
# New PersistenceImager Tests
@@ -155,18 +155,18 @@ def test_mixed_pairs():
155155

156156
class TestWeightFunctions:
157157
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})
159159
wf = persimgr.weight
160160
wf_params = persimgr.weight_params
161161
np.testing.assert_equal(wf(1, 0, **wf_params), 0)
162162

163-
persimgr = PersistenceImager(weight=persistence, weight_params={'n': 2})
163+
persimgr = PersistenceImager(weight=images_weights.persistence, weight_params={'n': 2})
164164
wf = persimgr.weight
165165
wf_params = persimgr.weight_params
166166
np.testing.assert_equal(wf(1, 0, **wf_params), 0)
167167

168168
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})
170170

171171
wf = persimgr.weight
172172
wf_params = persimgr.weight_params
@@ -198,7 +198,7 @@ def test_linear_ramp(self):
198198
np.testing.assert_equal(wf(1, 2, **wf_params), 2)
199199

200200
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})
202202

203203
wf = persimgr.weight
204204
wf_params = persimgr.weight_params
@@ -213,11 +213,11 @@ def test_persistence(self):
213213

214214
class TestKernelFunctions:
215215
def test_gaussian(self):
216-
kernel = gaussian
216+
kernel = images_kernels.gaussian
217217
kernel_params = {'mu':[1, 1], 'sigma':np.array([[1,0],[0,1]])}
218218
np.testing.assert_almost_equal(kernel(np.array([1]), np.array([1]), **kernel_params), 1/4, 8)
219219

220-
kernel = bvn_cdf
220+
kernel = images_kernels.bvn_cdf
221221
kernel_params = {'mu_x':1.0, 'mu_y':1.0, 'sigma_xx':1.0, 'sigma_yy':1.0, 'sigma_xy':0.0}
222222
np.testing.assert_almost_equal(kernel(np.array([1]), np.array([1]), **kernel_params), 1/4, 8)
223223

@@ -231,19 +231,19 @@ def test_gaussian(self):
231231
np.testing.assert_equal(kernel(np.array([1]), np.array([1]), **kernel_params), .375)
232232

233233
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)
236236

237237
def test_uniform(self):
238-
kernel = uniform
238+
kernel = images_kernels.uniform
239239
kernel_params={'width': 3, 'height': 1}
240240

241241
np.testing.assert_equal(kernel(np.array([-1]), np.array([-1]), mu=(0,0), **kernel_params), 0)
242242
np.testing.assert_equal(kernel(np.array([3]), np.array([1]), mu=(0,0), **kernel_params), 1)
243243
np.testing.assert_equal(kernel(np.array([5]), np.array([5]), mu=(0,0), **kernel_params), 1)
244244

245245
def test_sigma(self):
246-
kernel = gaussian
246+
kernel = images_kernels.gaussian
247247
kernel_params1 = {'sigma':np.array([[1,0],[0,1]])}
248248
kernel_params2 = {'sigma': [[1,0],[0,1]]}
249249
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

Comments
 (0)