def test_graphframe_to_literal(tau_profile_dir): """Sanity test a GraphFrame object with known data.""" gf = GraphFrame.from_tau(str(tau_profile_dir)) graph_literal = gf.to_literal() gf_literal = GraphFrame.from_literal(graph_literal) assert len(gf.graph) == len(gf_literal.graph)
def test_graphframe(tau_profile_dir): """Sanity test a GraphFrame object with known data.""" gf = GraphFrame.from_tau(str(tau_profile_dir)) for col in gf.dataframe.columns: if col in ("time (inc)", "time"): assert gf.dataframe[col].dtype == np.float64 elif col in ("line"): assert gf.dataframe[col].dtype == np.int64 elif col in ("name", "node"): assert gf.dataframe[col].dtype == np.object
def test_children(tau_profile_dir): gf = GraphFrame.from_tau(str(tau_profile_dir)) root = gf.graph.roots root_children = [ "MPI_Init()", "MPI_Comm_size()", "MPI_Comm_rank()", "MPI_Get_processor_name()", "MPI_Bcast()", "MPI_Reduce()", "MPI_Finalize()", ] # check if only one root node is created assert len(root) == 1 # check if root has right children #TODO: can be improved for child in root[0].children: assert child.frame["name"] in root_children
def test_tree(tau_profile_dir): """Sanity test a GraphFrame object with known data.""" gf = GraphFrame.from_tau(str(tau_profile_dir)) # check the tree for rank 0 output = ConsoleRenderer(unicode=True, color=False).render( gf.graph.roots, gf.dataframe, metric_column="time", precision=3, name_column="name", expand_name=False, context_column="file", rank=0, thread=0, depth=10000, highlight_name=False, colormap="RdYlGn", invert_colormap=False, ) assert "449.000 .TAU application" in output assert "4458.000 MPI_Finalize()" in output assert "218.000 MPI_Bcast()" in output # check the tree for rank 1 output = ConsoleRenderer(unicode=True, color=False).render( gf.graph.roots, gf.dataframe, metric_column="time", precision=3, name_column="name", expand_name=False, context_column="", rank=1, thread=0, depth=10000, highlight_name=False, colormap="RdYlGn", invert_colormap=False, ) assert "419.000 .TAU application" in output assert "4894.000 MPI_Finalize()" in output assert "333.000 MPI_Bcast()" in output