Skip to content

Commit ff8d96b

Browse files
committed
update unit tests
1 parent f90bef3 commit ff8d96b

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

pca/tests/test_pca.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def test_specifying(self):
5151
s=combination[2],
5252
marker=combination[0],
5353
alpha=combination[3],
54-
fontdict={'weight':'normal'},
54+
arrowdict={'weight':'normal'},
5555
color_arrow='blue',
5656
title='Demonstration of specifying colors, markers, alpha, and size per sample.',
5757
HT2=True,
@@ -148,26 +148,26 @@ def test_plot_combinations(self):
148148
X = pd.DataFrame(data=load_iris().data, columns=load_iris().feature_names, index=load_iris().target)
149149

150150
param_grid = {
151-
'n_components':[None, 0.01, 1, 0.95, 2, 100000000000],
151+
# 'n_components':[None, 0.01, 1, 0.95, 2, 100000000000],
152152
'row_labels':[None, [], y],
153153
'detect_outliers' : [None, 'ht2','spe'],
154-
'gradient' : [None, '#FFFFFF']
154+
# 'gradient' : [None, '#FFFFFF']
155155
}
156156

157157
allNames = param_grid.keys()
158158
combinations = it.product(*(param_grid[Name] for Name in allNames))
159159
combinations=list(combinations)
160-
160+
161161
for combination in combinations:
162162
model = pca(n_components=combination[0])
163-
model = pca(n_components=0.95)
163+
model = pca(n_components=3)
164164
model.fit_transform(X)
165-
assert model.plot()
166-
assert model.biplot(y=y, SPE=True, HT2=True, gradient=combination[3])
167-
assert model.biplot3d(y=y, SPE=True, HT2=True, gradient=combination[3])
168-
assert model.biplot(y=y, SPE=True, HT2=False, gradient=combination[3])
169-
assert model.biplot(y=y, SPE=False, HT2=True, gradient=combination[3])
170-
assert model.biplot(y=y, SPE=False, HT2=False, gradient=combination[3])
165+
assert model.plot(visible=False)
166+
assert model.biplot(labels=y, SPE=True, HT2=True, visible=False)
167+
assert model.biplot3d(labels=y, SPE=True, HT2=True, visible=False)
168+
assert model.biplot(labels=y, SPE=True, HT2=False, visible=False)
169+
assert model.biplot(labels=y, SPE=False, HT2=True, visible=False)
170+
assert model.biplot(labels=y, SPE=False, HT2=False, visible=False)
171171
assert model.results['PC'].shape[1]==model.n_components
172172

173173

0 commit comments

Comments
 (0)