Update error handling

Author: vgoliber

Exploring_Pegasus/biclique_embedding.py

@@ -27,7 +27,17 @@
  import matplotlib.pyplot as plt
 
 # Form the biclique
-N = int(sys.argv[1])
+args = sys.argv
+if len(args) <= 1:
+ print("\nNo input provided. Please provide an integer.")
+ exit(1)
+
+try:
+ N = int(args[1])
+except:
+ print("\nIncorrect input argument. Please provide an integer.\n")
+ exit(1)
+
 nodes1 = np.arange(N)
 nodes2 = [chr(i + ord('a')) for i in range(N)]
 B = nx.Graph()

Exploring_Pegasus/clique_embedding.py

@@ -25,7 +25,17 @@
  matplotlib.use("agg")
  import matplotlib.pyplot as plt
 
-N = int(sys.argv[1])
+args = sys.argv
+if len(args) <= 1:
+ print("\nNo input provided. Please provide an integer.")
+ exit(1)
+
+try:
+ N = int(args[1])
+except:
+ print("\nIncorrect input argument. Please provide an integer.\n")
+ exit(1)
+
 G = nx.complete_graph(N)
 
 fig, axes = plt.subplots(nrows=1, ncols=3, figsize=(12, 6))