def test_from_dataframe_with_transforms_and_fillna(dataframe): df = dataframe.df def key1(x): return x**2 def key2(x): return (x < 5) * x def key3(x): return (x > 0) * x h = Hypergraph.from_dataframe(df) assert "A" in h.edges["a"] assert "A" not in h.edges["b"] h = Hypergraph.from_dataframe(df, fillna=1) assert "A" in h.edges["b"] h = Hypergraph.from_dataframe(df, transforms=[key1, key2]) assert "A" in h.edges["c"] assert "C" not in h.edges["b"] h = Hypergraph.from_dataframe(df, transforms=[key2, key3]) assert "C" in h.edges["b"] h = Hypergraph.from_dataframe(df, transforms=[key3, key1], key=key2) assert "A" not in h.edges["a"] assert "B" in h.edges["b"] assert "C" not in h.edges["c"] assert "C" in h.edges["a"]
def test_from_dataframe_with_key(): M = np.array([[5, 0, 7, 2], [6, 8, 1, 1], [2, 5, 1, 9]]) index = ["A", "B", "C"] columns = ["a", "b", "c", "d"] df = pd.DataFrame(M, index=index, columns=columns) h = Hypergraph.from_dataframe(df, key=lambda x: x > 4) assert "A" in h.edges["a"] assert "C" not in h.edges["a"]
def test_from_dataframe_with_key(): M = np.array([[5, 0, 7, 2], [6, 8, 1, 1], [2, 5, 1, 9]]) index = ['A', 'B', 'C'] columns = ['a', 'b', 'c', 'd'] df = pd.DataFrame(M, index=index, columns=columns) h = Hypergraph.from_dataframe(df, key=lambda x: x > 4) assert 'A' in h.edges['a'] assert 'C' not in h.edges['a']
def test_from_dataframe(): M = np.array([[1, 1, 0, 0], [0, 1, 1, 0], [1, 0, 1, 0]]) index = ["A", "B", "C"] columns = ["a", "b", "c", "d"] df = pd.DataFrame(M, index=index, columns=columns) h = Hypergraph.from_dataframe(df) assert "b" in h.edges assert "d" not in h.edges assert "C" in h.edges["a"]
def test_from_dataframe(): M = np.array([[1, 1, 0, 0], [0, 1, 1, 0], [1, 0, 1, 0]]) index = ['A', 'B', 'C'] columns = ['a', 'b', 'c', 'd'] df = pd.DataFrame(M, index=index, columns=columns) h = Hypergraph.from_dataframe(df) assert 'b' in h.edges assert 'd' not in h.edges assert 'C' in h.edges['a']
def test_hypergraph_from_dataframe(lesmis): df = lesmis.hypergraph.dataframe() H = Hypergraph.from_dataframe(df) assert H.shape == (40, 8) assert H.size(3) == 8 assert H.degree("JA") == 3