Fix compile time evaluation for capitalize

Author: AbdelrahmanKhaledd

src/lpython/semantics/python_comptime_eval.h

@@ -655,21 +655,17 @@ struct PythonIntrinsicProcedures {
 
     static ASR::expr_t *eval__lpython_str_capitalize(Allocator &al, const Location &loc, Vec<ASR::expr_t *> &args) {
         LFORTRAN_ASSERT(ASRUtils::all_args_evaluated(args));
-        if (args.size() != 1) {
-            throw SemanticError("_lpython_str_capitalize() takes exactly one arguments (" +
-                std::to_string(args.size()) + " given)", loc);
-        }
-        ASR::expr_t *arg1 = args[0];
-        ASR::ttype_t *arg1_type = ASRUtils::expr_type(arg1);
-        if (ASRUtils::is_character(*arg1_type)) {
-            std::string val = ASR::down_cast<ASR::StringConstant_t>(arg1)->m_s;
-            ASR::ttype_t *type = ASRUtils::TYPE(ASR::make_Character_t(al, loc,
-                                        1, 1, nullptr, nullptr, 0));
-            ASR::ttype_t *res_type = ASRUtils::TYPE(ASR::make_StringConstant_t(al, loc, s2c(al, ""), type));
-            return ASR::down_cast<ASR::expr_t>(ASR::make_StringConstant_t(al, loc, s2c(al, val),  res_type));
-        } else {
-            throw SemanticError("Only string from arguments.", loc);
-        }
+        if (args.size() != 0) {
+            throw SemanticError("str.capitalize() takes no arguments", loc);
+        }
+        ASR::expr_t *arg = args[0];
+        ASR::ttype_t *arg_type = ASRUtils::expr_type(arg);
+        std::string val = ASR::down_cast<ASR::StringConstant_t>(arg)->m_s;
+        val = std::toupper(val[0]);
+        ASR::ttype_t *type = ASRUtils::TYPE(ASR::make_Character_t(al, loc,
+                                    1, 1, nullptr, nullptr, 0));
+        ASR::ttype_t *res_type = ASRUtils::TYPE(ASR::make_StringConstant_t(al, loc, s2c(al, ""), type));
+        return ASR::down_cast<ASR::expr_t>(ASR::make_StringConstant_t(al, loc, s2c(al, val),  res_type));
     }
 
     static ASR::expr_t *eval__lpython_str_lower(Allocator &al, const Location &loc, Vec<ASR::expr_t *> &args) {