def test_inputs():
graphs, labels = generate_data()
mugs = mug2vec(omnibus_components=-1)
with pytest.raises(ValueError):
mugs.fit(graphs)
mugs = mug2vec(cmds_components=-1)
with pytest.raises(ValueError):
mugs.fit(graphs)
mugs = mug2vec(omnibus_n_elbows=-1)
with pytest.raises(ValueError):
mugs.fit(graphs)
mugs = mug2vec(cmds_n_elbows=-1)
with pytest.raises(ValueError):
mugs.fit(graphs)
def test_inputs(self):
graphs, labels = generate_data()
mugs = mug2vec(omnibus_components=-1, svd_seed=1)
with self.assertRaises(ValueError):
mugs.fit(graphs)
mugs = mug2vec(cmds_components=-1, svd_seed=1)
with self.assertRaises(ValueError):
mugs.fit(graphs)
mugs = mug2vec(omnibus_n_elbows=-1, svd_seed=1)
with self.assertRaises(ValueError):
mugs.fit(graphs)
mugs = mug2vec(cmds_n_elbows=-1, svd_seed=1)
with self.assertRaises(ValueError):
mugs.fit(graphs)
def test_mug2vec():
graphs, labels = generate_data()
mugs = mug2vec(pass_to_ranks=None)
xhat = mugs.fit_transform(graphs)
gmm = GaussianCluster(5)
gmm.fit(xhat, labels)
assert_equal(gmm.n_components_, 2)
def test_mug2vec(self):
graphs, labels = generate_data()
mugs = mug2vec(pass_to_ranks=None, svd_seed=1)
xhat = mugs.fit_transform(graphs)
gmm = GaussianCluster(5)
gmm.fit(xhat, labels)
self.assertEqual(gmm.n_components_, 2)