(self, num_nodes=3, sample_method="chain")
| 37 | return self.sample_subgraph(tool_number, sample_method=method) |
| 38 | |
| 39 | def sample_subgraph(self, num_nodes=3, sample_method="chain"): |
| 40 | seed_node = random.choice(list(self.graph.nodes)) |
| 41 | if sample_method == "single": |
| 42 | sub_G = nx.DiGraph() |
| 43 | sub_G.add_node(seed_node) |
| 44 | return sub_G |
| 45 | elif sample_method == "chain": |
| 46 | return self.sample_subgraph_chain(seed_node, num_nodes) |
| 47 | elif sample_method == "dag": |
| 48 | return self.sample_subgraph_dag(seed_node, num_nodes) |
| 49 | else: |
| 50 | raise ValueError("Invalid sample method") |
| 51 | |
| 52 | def sample_subgraph_chain(self, seed_node, num_nodes): |
| 53 | # Create a list to store the sub-graph nodes |
no test coverage detected