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)