Added built in method upper() for string (#1714)

Author: AtmegaBuzz

integration_tests/test_str_attributes.py

@@ -21,6 +21,12 @@ def lower():
     assert "DDd12Vv" .lower() == "ddd12vv"
     assert "".lower() == ""
 
+def upper():
+    s: str
+    s = "AaaaAABBbbbbBB!@12223BN"
+    assert s.upper() == "AAAAAABBBBBBBB!@12223BN"
+    assert "DDd12Vv".upper() == "DDD12VV"
+    assert "".upper() == ""
 
 def strip():
     s: str
@@ -179,6 +185,7 @@ def partition():
 def check():
     capitalize()
     lower()
+    upper()
     strip()
     swapcase()
     find()

src/lpython/semantics/python_ast_to_asr.cpp

@@ -5773,6 +5773,16 @@ class BodyVisitor : public CommonVisitor<BodyVisitor> {
             arg.loc = loc;
             arg.m_value = s_var;
             fn_args.push_back(al, arg);
+        } else if (attr_name == "upper") {
+            if (args.size() != 0) {
+                throw SemanticError("str.upper() takes no arguments",
+                    loc);
+            }
+            fn_call_name = "_lpython_str_upper";
+            ASR::call_arg_t arg;
+            arg.loc = loc;
+            arg.m_value = s_var;
+            fn_args.push_back(al, arg);
         } else if (attr_name == "find") {
             if (args.size() != 1) {
                 throw SemanticError("str.find() takes one argument",
@@ -6062,6 +6072,16 @@ class BodyVisitor : public CommonVisitor<BodyVisitor> {
                     i = tolower(i);
                 }
             }
+        } else if (attr_name == "upper") {
+            if (args.size() != 0) {
+                throw SemanticError("str.upper() takes no arguments",
+                        loc);
+            }
+            for (auto &i : s_var) {
+                if (i >= 'a' && i<= 'z') {
+                    i = toupper(i);
+                }
+            }
         } else if (attr_name == "find") {
             if (args.size() != 1) {
                 throw SemanticError("str.find() takes one arguments",

src/lpython/semantics/python_comptime_eval.h

@@ -72,6 +72,7 @@ struct PythonIntrinsicProcedures {
             // for evaluation.
             {"_lpython_str_capitalize", {m_builtin, &not_implemented}},
             {"_lpython_str_lower", {m_builtin, &not_implemented}},
+            {"_lpython_str_upper", {m_builtin, &not_implemented}},
             {"_lpython_str_find", {m_builtin, &not_implemented}},
             {"_lpython_str_rstrip", {m_builtin, &not_implemented}},
             {"_lpython_str_lstrip", {m_builtin, &not_implemented}},

src/runtime/lpython_builtin.py

@@ -689,12 +689,25 @@ def _lpython_str_lower(x: str) -> str:
     res = ""
     i:str
     for i in x:
-        if ord('A') <= ord(i) and ord('Z') >= ord(i):
+        if ord('A') <= ord(i) and ord(i) <= ord('Z'):
             res += chr(ord(i) +32)
         else:
             res += i
     return res
 
+@overload
+def _lpython_str_upper(x: str) -> str:
+    res: str
+    res = ""
+    i:str
+    for i in x:
+        if ord('a') <= ord(i) and ord(i) <= ord('z'):
+            res += chr(ord(i) -32)
+        else:
+            res += i
+    return res
+
+
 @overload
 def _lpython_str_find(s: str, sub: str) -> i32:
     s_len :i32; sub_len :i32; flag: bool; _len: i32;

tests/reference/asr-array_01_decl-39cf894.json

@@ -6,7 +6,7 @@
     "outfile": null,
     "outfile_hash": null,
     "stdout": "asr-array_01_decl-39cf894.stdout",
-    "stdout_hash": "b2131e7fac5ffaeae5379d22cbe0362cc22d8b6ca62e9f49e7d4f780",
+    "stdout_hash": "57442cc5bdf2788dfde4f41fa74c886757651c0dc6a62b4f3e0ec745",
     "stderr": null,
     "stderr_hash": null,
     "returncode": 0

tests/reference/asr-array_01_decl-39cf894.stdout

@@ -6,7 +6,7 @@
     "outfile": null,
     "outfile_hash": null,
     "stdout": "asr-array_02_decl-e8f6874.stdout",
-    "stdout_hash": "2adb7f466d09d9c6911977465fb6736be6b54a076074d9827688c7da",
+    "stdout_hash": "0744133e16cf5b9274d2e3172c2cd9efb0f25962023b7bb240f2e753",
     "stderr": null,
     "stderr_hash": null,
     "returncode": 0

tests/reference/asr-array_02_decl-e8f6874.json

@@ -6,7 +6,7 @@
     "outfile": null,
     "outfile_hash": null,
     "stdout": "asr-bindc_02-bc1a7ea.stdout",
-    "stdout_hash": "90ed89ce4e51d48a2f3676b335b22ed329b262136a0988ecd3f6d4b2",
+    "stdout_hash": "253e33fc1172553b2973f9298c8db9d7d1cca284e77c37cb48ef1cbd",
     "stderr": null,
     "stderr_hash": null,
     "returncode": 0