Update calculator.py

diyorbekibragimov · Jun 5, 2024 · a880e0a · a880e0a
1 parent 1c3782d
commit a880e0a
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/calculator.py b/calculator.py
@@ -31,17 +31,20 @@
     '3'
 """
 def add(s: str) -> str:
-    sum = 0
-    if (s == ""):
-        return "0"
-    s = s.replace('\n', ',')
-    L = s.split(",")
-    for i in range(len(L)):
-        if "." in L[i]:
-            sum += float(L[i])
-        else:
-            sum += int(L[i])
-    return str(sum)
+  if (s[-1] == ',' or s[-1] == '\n'){
+    return 'Number expected but EOF found.'
+  }
+  sum = 0
+  if (s == ""):
+    return "0"
+  s = s.replace('\n', ',')
+  L = s.split(",")
+  for i in range(len(L)):
+      if "." in L[i]:
+          sum += float(L[i])
+      else:
+          sum += int(L[i])
+  return str(sum)