Handle inputs like "2 +" in StackPostfixNotation (TheAlgorithms#4262)

vil02 · web-flow · commit e5c7a08874a6 · 2023-07-26T14:11:28.000+03:00
diff --git a/src/main/java/com/thealgorithms/others/StackPostfixNotation.java b/src/main/java/com/thealgorithms/others/StackPostfixNotation.java
@@ -16,6 +16,9 @@ public static int postfixEvaluate(final String exp) {
             if (tokens.hasNextInt()) {
                 s.push(tokens.nextInt()); // If int then push to stack
             } else { // else pop top two values and perform the operation
+                if (s.size() < 2) {
+                    throw new IllegalArgumentException("exp is not a proper postfix expression (too few arguments).");
+                }
                 int num2 = s.pop();
                 int num1 = s.pop();
                 String op = tokens.next();
diff --git a/src/test/java/com/thealgorithms/others/StackPostfixNotationTest.java b/src/test/java/com/thealgorithms/others/StackPostfixNotationTest.java
@@ -30,4 +30,14 @@ public void testIfEvaluateThrowsExceptionForInproperInput() {
     public void testIfEvaluateThrowsExceptionForInputWithUnknownOperation() {
         assertThrows(IllegalArgumentException.class, () -> StackPostfixNotation.postfixEvaluate("3 3 !"));
     }
+
+    @Test
+    public void testIfEvaluateThrowsExceptionForInputWithTooFewArgsA() {
+        assertThrows(IllegalArgumentException.class, () -> StackPostfixNotation.postfixEvaluate("+"));
+    }
+
+    @Test
+    public void testIfEvaluateThrowsExceptionForInputWithTooFewArgsB() {
+        assertThrows(IllegalArgumentException.class, () -> StackPostfixNotation.postfixEvaluate("2 +"));
+    }
 }
