def test_ggd_transform_undirected(X, X_res, method):
with warnings.catch_warnings():
warnings.filterwarnings("ignore", message="Methods .*")
ggd = GraphGeodesicDistance(directed=False, method=method)
X_ft = ggd.fit_transform(X)
assert_almost_equal(X_ft, X_res)
def test_ggd_transform():
X_ggd_res = np.array([[[0., 1., 3., 7., np.inf], [1., 0., 4., 8., np.inf],
[3., 4., 0., 4., np.inf], [7., 8., 4., 0., np.inf],
[np.inf, np.inf, np.inf, np.inf, 0.]],
[[0., 1., 2., 6., np.inf], [1., 0., 1., 5., np.inf],
[2., 1., 0., 4., np.inf], [6., 5., 4., 0., np.inf],
[np.inf, np.inf, np.inf, np.inf, 0.]]])
ggd = GraphGeodesicDistance()
assert_almost_equal(ggd.fit_transform(X_ggd), X_ggd_res)
def test_parallel_ggd_transform():
X = X_ggd[0][0]
ggd = GraphGeodesicDistance(n_jobs=1)
ggd_parallel = GraphGeodesicDistance(n_jobs=2)
with warnings.catch_warnings():
warnings.filterwarnings("ignore", message="Methods .*")
assert_almost_equal(ggd.fit_transform(X),
ggd_parallel.fit_transform(X))
def test_ggd_fit_transform_plot():
X = X_ggd[0][0]
with warnings.catch_warnings():
warnings.filterwarnings("ignore", message="Methods .*")
GraphGeodesicDistance().fit_transform_plot(X, sample=0)
def test_ggd_not_fitted():
ggd = GraphGeodesicDistance()
with pytest.raises(NotFittedError):
ggd.transform(X_ggd)
def test_parallel_ggd_transform():
ggd = GraphGeodesicDistance(n_jobs=1)
ggd_parallel = GraphGeodesicDistance(n_jobs=2)
assert_almost_equal(ggd.fit_transform(X_ggd),
ggd_parallel.fit_transform(X_ggd))
def test_ggd_fit_transform_plot():
GraphGeodesicDistance().fit_transform_plot(X_ggd, sample=0)
def test_ggd_transform(X, X_res, method):
ggd = GraphGeodesicDistance(directed=False, method=method)
assert_almost_equal(ggd.fit_transform(X), X_res)
def test_ggd_fit_transform_plot():
X = X_ggd[0][0]
GraphGeodesicDistance().fit_transform_plot(X, sample=0)
