Jeon193 06arg

06_argparser/submissions/jeon193.py

@@ -0,0 +1,37 @@
+import numpy as np
+
+
+def afunc(array1, array2, mode="+", pow=2):
+    assert array1.shape == array2.shape, "size mismatch"
+    assert type(pow) == int, "pow should be type int"
+
+    if mode == "+":
+        result = array1 + array2
+    elif mode == "-":
+        result = array1 - array2
+    else:
+        raise ValueError(f"mode is either '+' or '-' but got {mode}")
+
+    result = result ** pow
+    return result
+
+
+if __name__ == "__main__":
+    import argparse
+
+    parser = argparse.ArgumentParser(description="This is tutorial for argparse.")  # noqa: E501
+    # add your code here
+    parser.add_argument('mode', metavar='mode', type=str, help='a mode string', default='+')  # noqa: E501
+    parser.add_argument('pow', metavar='pow', type=int, help='an int pow', default=2)  # noqa: E501
+    parser.add_argument('print_hello', action='store_true')
+
+    # add your code here
+    args = parser.parse_args()
+    print(args)
+
+    a = np.array([1, 2, 3])
+    b = np.array([4, 5, 6])
+    result = afunc(a, b, args.mode, args.pow)
+    print(result)
+    if args.print_hello:
+        print("Hello!")