def test_graphframe(calc_pi_callgrind_dot):
"""Sanity test a GraphFrame object with known data."""
gf = GraphFrame()
gf.from_gprof_dot(str(calc_pi_callgrind_dot))
assert len(gf.dataframe.groupby('name')) == 105
def test_graphframe(calc_pi_callgrind_dot):
"""Sanity test a GraphFrame object with known data."""
gf = GraphFrame.from_gprof_dot(str(calc_pi_callgrind_dot))
assert len(gf.dataframe.groupby("name")) == 105
for col in gf.dataframe.columns:
if col in ("time (inc)", "time"):
assert gf.dataframe[col].dtype == np.float64
elif col in ("name", "module", "node"):
assert gf.dataframe[col].dtype == np.object
