From dd24776149e54867f9ea4b6f6b91c71c048142d2 Mon Sep 17 00:00:00 2001 From: Joris van Rantwijk Date: Sat, 11 Mar 2023 22:00:38 +0100 Subject: [PATCH] Add missing import statements --- tests/generate/hardcard.py | 3 ++- tests/generate/triangles.py | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/generate/hardcard.py b/tests/generate/hardcard.py index c7785ff..da897f5 100644 --- a/tests/generate/hardcard.py +++ b/tests/generate/hardcard.py @@ -25,10 +25,12 @@ The graph is constructed so that vertices 1 - 4*K form a complete subgraph. For 1 <= I <= 2*K, vertex (2*I-1) is joined to vertex (4*K+I). """ +import sys import argparse def main(): + """Main program.""" parser = argparse.ArgumentParser() parser.description = "Generate a difficult graph" @@ -59,4 +61,3 @@ def main(): if __name__ == "__main__": main() - diff --git a/tests/generate/triangles.py b/tests/generate/triangles.py index 87f2c6e..39e4d30 100644 --- a/tests/generate/triangles.py +++ b/tests/generate/triangles.py @@ -41,10 +41,12 @@ Number of vertices: N = 3*K Number of edges: M = 3*K + C*(K-1) """ +import sys import argparse def main(): + """Main program.""" parser = argparse.ArgumentParser() parser.description = (