()
| 10 | |
| 11 | |
| 12 | def test_LINE(): |
| 13 | graph = nx.read_edgelist( |
| 14 | str(TEST_GRAPH_PATH), |
| 15 | create_using=nx.DiGraph(), |
| 16 | nodetype=None, |
| 17 | data=[("weight", int)], |
| 18 | ) |
| 19 | |
| 20 | model = LINE(graph, embedding_size=4, order="second") |
| 21 | model.train(batch_size=2, epochs=1, verbose=0) |
| 22 | embeddings = model.get_embeddings() |
| 23 | assert len(embeddings) == graph.number_of_nodes() |
| 24 | assert all(len(vector) == 4 for vector in embeddings.values()) |
| 25 | |
| 26 | |
| 27 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected