New test cases at resources + input/output

Signed-off-by: Flavia Beo <flavia.beo@ibm.com>

Author: flaviabeo

tests/resources/prompts/granite/input/67.txt

@@ -1,10 +1,11 @@
-translate code from cobol to java
+System:
+Solve the following coding problem. Wrap your code answer using ```
 
 
-Input:
+Question:
+Translate code from cobol to java:
+
 DEMO-TASKA.
- 
- * This is prompt example 1 *  
 MOVE 'HELLO WORLD' TO VA-TAB  
 MOVE '45' TO CA-PAYMENT  
 IF MAN-BOOKER IS EQUAL TO 24
@@ -30,223 +31,5 @@ ENDIF
  DISPLAY AGE
  EXIT.
 
-## Variable Map:
-VA-TAB gettermapping recordBook.getRetailName().getTabValue()
-VA-TAB settermapping recordBook.getRetailName().setTabValue(val)
-VA-TAB PIC X(20) JavaType String
-MAN-BOOKER gettermapping markDanson.getPost().getManBooker() 
-MAN-BOOKER settermapping markDanson.getPost().setManBooker(val) 
-MAN-BOOKER PIC 9(3) JavaType int
-CA-PAYMENT gettermapping this.getCaPayment()
-CA-PAYMENT settermapping this.setCaPayment(val)
-CA-PAYMENT PIC 9(2) JavaType int
-PY-RES-INT gettermapping this.getPyResInt()
-PY-RES-INT settermapping this.setPyResInt(val)
-PY-RES-INT PIC 99 JavaType int
-SHAP-VALUE gettermapping this.retail.getShapValue()
-SHAP-VALUE settermapping this.retail.setShapValue(val)
-SHAP-VALUE PIC X(4) JavaType String
-SAN-NUM gettermapping sanNum
-SAN-NUM settermapping sanNum
-SAN-NUM PIC 9(2) JavaType int
-AGE gettermapping spa.getInt("age")
-AGE settermapping spa.setInt("age", val)
-AGE PIC 9(2) JavaType int
-TRAN-RATE-INTERVAL, markDanson.tranInterval(reportVal)
-
-## Class Map:
-void demoTaskA(int sanNum){
-    bookShop markDanson =  new BookShop();
-}
-
-Output:
-void demoTaskA(int sanNum) {
-    bookShop markDanson =  new BookShop();
-    /* This is prompt example 1 */ 
-    recordBook.getRetailName().setTabValue("HELLO WORLD");
-    this.setCaPayment(Integer.parseInt("45"));        
-   
-    if (markDanson.getPost().getManBooker() == 24) {
-    try {
-        String sql = "INSERT INTO ADDR(NUMBER, NAME)values(?, ?)";
-        PreparedStatement ps = JdbcConnection.connection.prepareStatement(sql);
-        ps.setInt(1, sanNum);
-        ps.setString(2, this.retail.getShapValue());
-        ps.executeUpdate();
-        ps.close();
-    }
-    catch(SQLException exception) {
-        markDanson.tranInterval(reportVal);
-        return;
-    }
-    }
-    try{
-        ResultSet rs =  JdbcConnection.connection.createStatement().executeQuery("SELECT IDENTITY_VAL_LOCAL() FROM SYSIBM.SYSDUMMY1");
-        rs.next();
-        sanNum = rs.getInt(1);
-    }
-    catch(SQLException exception){
-        System.out.println(exception);
-        return;
-    }
-    this.setPyResInt(sanNum);  
-    markDanson.getPost().setManBooker(25);
-    spa.setInt("age", 20);
-    System.out.println( spa.getInt("age") );
-}
-
-Input:
-BUZZ-VAL. 
-* This is prompt example 2 * 
- IF CAN-BANK-VAL IS EQUAL TO 87
- EXEC SQL
-   INSERT INTO BANK
-             ( NUM,
-               LOCATION)
-      VALUES ( DEFAULT,
-               :SYS-GERM)
- END-EXEC
- IF SQLCODE NOT EQUAL 0   
-   MOVE '10' TO CA-SUM-ASSURED
-   PERFORM PRINT-FUNCTION
-   EXEC STATEMENT2 END-EXEC
- END-IF
-END-IF
-EXIT.
-
-## Variable Map:
-SYS-GERM gettermapping sysGerm
-SYS-GERM settermapping sysGerm
-SYS-GERM PIC X(5) JavaType String
-CAN-BANK-VAL gettermapping bankVar.getCanBankVal()
-CAN-BANK-VAL settermapping bankVar.setCanBankVal(val)
-CAN-BANK-VAL 9(3) JavaType int
-CA-SUM-ASSURED gettermapping this.getCaSumAssured()
-CA-SUM-ASSURED settermapping this.setCaSumAssured(val)
-CA-SUM-ASSURED PIC 99 JavaType int
-PRINT-FUNCTION, this.printFunction(errReport)
-
-## Class Map:
-void buzzVal(String caRhNum){        
-    int poll = 0;
-}
-
-Output:
-void buzzVal(String caRhNum){
-    int poll = 0;
-   /* This is prompt example 2 */
-   if (bankVar.getCanBankVal() == 87){
-     try {
-        String sql = "INSERT INTO BANK(LOCATION)values(?)";
-        PreparedStatement ps = JdbcConnection.connection.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
-        ps.setString(1, sysGerm);
-        ps.executeUpdate();
-        ps.close();
-    }
-    catch(SQLException exception) {
-        this.setCaSumAssured(Integer.parseInt("10"));     
-        this.printFunction(errReport);
-        return;
-     } 
-   }
-}
-
-Input:
-           IF WS-DISPLAY-LINE = "* MQPUTSUB V1.0 " AND
-              CA-RETURN-CODE = 0 AND
-              WS-DEBUG = " " AND
-              WS-COMM-QUEUE = "COMM BLOCK DATA" AND
-              B100-NOT-RAN AND
-              B117-NOT-RAN AND
-              W500-RAN THEN
-               DISPLAY "TEST 3 PASSED"
-           ELSE
-               DISPLAY "TEST 3 FAILED"
-           END-IF
-           INITIALIZE WS-COMM-BLOCK
-           INITIALIZE WS-DISPLAY-LINE
-           INITIALIZE WS-CONFIG-PARAMETERS
-           INITIALIZE CA-MQPUTSUB-PARM
-               SET DISCONNECT-ACTION TO TRUE
-               SET TOGGLE-ERROR-FLAG TO TRUE
-               MOVE "COMM BLOCK DATA" TO CA-COMM-BLOCK
-               MOVE "QUEUE DATA" TO WS-QUEUE
-           INITIALIZE SECTIONS-RUN
-               PERFORM B000-INITIALISE.
-           IF WS-DISPLAY-LINE = "* MQPUTSUB V1.0 " AND
-              CA-RETURN-CODE = 0 AND
-              WS-DEBUG = " " AND
-              WS-COMM-QUEUE = "COMM BLOCK DATA" AND
-              B100-NOT-RAN AND
-              B117-NOT-RAN AND
-              W500-RAN THEN
-               DISPLAY "TEST 4 PASSED"
-           ELSE
-               DISPLAY "TEST 4 FAILED"
-           END-IF.
-
-## Variable Map:
-WS-QUEUE gettermapping wsConfigParameters.getWsQueue()
-WS-QUEUE settermapping wsConfigParameters.setWsQueue(val)
-WS-QUEUE PIC X(00048) JavaType String
-WS-DISPLAY-LINE gettermapping wsDisplayLine.getBytes()
-WS-DISPLAY-LINE settermapping wsDisplayLine.setBytes(val)
-WS-DISPLAY-LINE GROUP JavaType WsDisplayLine
-WS-DEBUG gettermapping wsCommBlock.getWsDebug()
-WS-DEBUG settermapping wsCommBlock.setWsDebug(val)
-WS-DEBUG PIC X(00001) JavaType char
-WS-CONFIG-PARAMETERS gettermapping wsConfigParameters.getBytes()
-WS-CONFIG-PARAMETERS settermapping wsConfigParameters.setBytes(val)
-WS-CONFIG-PARAMETERS GROUP JavaType WsConfigParameters
-WS-COMM-QUEUE gettermapping wsCommBlock.getWsCommQueue()
-WS-COMM-QUEUE settermapping wsCommBlock.setWsCommQueue(val)
-WS-COMM-QUEUE PIC X(00048) JavaType String
-WS-COMM-BLOCK gettermapping wsCommBlock.getBytes()
-WS-COMM-BLOCK settermapping wsCommBlock.setBytes(val)
-WS-COMM-BLOCK GROUP JavaType WsCommBlock
-W500-RAN gettermapping sectionsRun.isW500Ran()
-W500-RAN settermapping sectionsRun.setW500Ran()
-W500-RAN 88 JavaType boolean
-TOGGLE-ERROR-FLAG gettermapping testFlags.isToggleErrorFlag()
-TOGGLE-ERROR-FLAG settermapping testFlags.setToggleErrorFlag()
-TOGGLE-ERROR-FLAG 88 JavaType boolean
-TEST-ERROR-FLAG gettermapping testFlags.getTestErrorFlag()
-TEST-ERROR-FLAG settermapping testFlags.setTestErrorFlag(val)
-TEST-ERROR-FLAG PIC 9 JavaType int
-SECTIONS-RUN gettermapping sectionsRun.getBytes()
-SECTIONS-RUN settermapping sectionsRun.setBytes(val)
-SECTIONS-RUN GROUP JavaType SectionsRun
-DISCONNECT-ACTION gettermapping caMqputsubParm.getCaHeader().isDisconnectAction()
-DISCONNECT-ACTION settermapping caMqputsubParm.getCaHeader().setDisconnectAction()
-DISCONNECT-ACTION 88 JavaType boolean
-CA-RETURN-CODE gettermapping caMqputsubParm.getCaHeader().getCaReturnCode()
-CA-RETURN-CODE settermapping caMqputsubParm.getCaHeader().setCaReturnCode(val)
-CA-RETURN-CODE PIC 9(00004) JavaType int
-CA-MQPUTSUB-PARM gettermapping caMqputsubParm.getBytes()
-CA-MQPUTSUB-PARM settermapping caMqputsubParm.setBytes(val)
-CA-MQPUTSUB-PARM GROUP JavaType CaMqputsubParm
-CA-COMM-BLOCK gettermapping caMqputsubParm.getCaHeader().getCaCommBlock()
-CA-COMM-BLOCK settermapping caMqputsubParm.getCaHeader().setCaCommBlock(val)
-CA-COMM-BLOCK PIC X(00057) JavaType String
-CA-ACTION gettermapping caMqputsubParm.getCaHeader().getCaAction()
-CA-ACTION settermapping caMqputsubParm.getCaHeader().setCaAction(val)
-CA-ACTION PIC X(00004) JavaType String
-B117-NOT-RAN gettermapping sectionsRun.isB117NotRan()
-B117-NOT-RAN settermapping sectionsRun.setB117NotRan()
-B117-NOT-RAN 88 JavaType boolean
-B100-NOT-RAN gettermapping sectionsRun.isB100NotRan()
-B100-NOT-RAN settermapping sectionsRun.setB100NotRan()
-B100-NOT-RAN 88 JavaType boolean
-B000-INITIALISE, b000Initialise(caMqputsubParm, flErrorFlag, sectionsRun, testFlags, wsCommBlock, wsConfigParameters, wsDisplayLine)
-
-## Class Map:
-public void testPara(){
-    CaMqputsubParm caMqputsubParm = new CaMqputsubParm();
-    FlErrorFlag flErrorFlag = new FlErrorFlag();
-    SectionsRun sectionsRun = new SectionsRun();
-    TestFlags testFlags = new TestFlags();
-    WsCommBlock wsCommBlock = new WsCommBlock();
-    WsConfigParameters wsConfigParameters = new WsConfigParameters();
-    WsDisplayLine wsDisplayLine = new WsDisplayLine();
-}
 
+Answer:

tests/resources/prompts/granite/input/68.txt

@@ -0,0 +1,27 @@
+System:
+Solve the following coding problem. Wrap your code answer using ```
+
+
+Question:
+Translate code from cobol to java:
+
+BUZZ-VAL. 
+IF CAN-BANK-VAL IS EQUAL TO 87
+ EXEC SQL
+   INSERT INTO BANK
+             ( NUM,
+               LOCATION)
+      VALUES ( DEFAULT,
+               :SYS-GERM)
+ END-EXEC
+ IF SQLCODE NOT EQUAL 0   
+   MOVE '10' TO CA-SUM-ASSURED
+   PERFORM PRINT-FUNCTION
+   EXEC STATEMENT2 END-EXEC
+ END-IF
+END-IF
+EXIT.
+
+
+Answer:
+

tests/resources/prompts/granite/input/69.txt

@@ -0,0 +1,41 @@
+System:
+Solve the following coding problem. Wrap your code answer using ```
+
+
+Question:
+Translate code from cobol to java:
+DIF WS-DISPLAY-LINE = "* MQPUTSUB V1.0 " AND
+              CA-RETURN-CODE = 0 AND
+              WS-DEBUG = " " AND
+              WS-COMM-QUEUE = "COMM BLOCK DATA" AND
+              B100-NOT-RAN AND
+              B117-NOT-RAN AND
+              W500-RAN THEN
+               DISPLAY "TEST 3 PASSED"
+           ELSE
+               DISPLAY "TEST 3 FAILED"
+           END-IF
+           INITIALIZE WS-COMM-BLOCK
+           INITIALIZE WS-DISPLAY-LINE
+           INITIALIZE WS-CONFIG-PARAMETERS
+           INITIALIZE CA-MQPUTSUB-PARM
+               SET DISCONNECT-ACTION TO TRUE
+               SET TOGGLE-ERROR-FLAG TO TRUE
+               MOVE "COMM BLOCK DATA" TO CA-COMM-BLOCK
+               MOVE "QUEUE DATA" TO WS-QUEUE
+           INITIALIZE SECTIONS-RUN
+               PERFORM B000-INITIALISE.
+           IF WS-DISPLAY-LINE = "* MQPUTSUB V1.0 " AND
+              CA-RETURN-CODE = 0 AND
+              WS-DEBUG = " " AND
+              WS-COMM-QUEUE = "COMM BLOCK DATA" AND
+              B100-NOT-RAN AND
+              B117-NOT-RAN AND
+              W500-RAN THEN
+               DISPLAY "TEST 4 PASSED"
+           ELSE
+               DISPLAY "TEST 4 FAILED"
+           END-IF.
+
+
+Answer:

tests/resources/prompts/granite/output-20b-code-instruct/512/64.txt

@@ -0,0 +1,66 @@
+System:
+Solve the following coding problem. Wrap your code answer using ```
+
+
+Question:
+Translate the following cobol code to python:
+MOVE 'HELLO WORLD' TO VA-TAB  
+MOVE '45' TO CA-PAYMENT  
+IF MAN-BOOKER IS EQUAL TO 24
+ EXEC SQL
+   INSERT INTO ADDR
+             ( NUMBER,
+               NAME)
+      VALUES ( :SAN-NUM,
+               :SHAP-VALUE)              
+ END-EXEC
+ IF SQLCODE NOT EQUAL 0
+   PERFORM TRAN-RATE-INTERVAL
+   EXEC STATEMENT1 END-EXEC
+ END-IF
+ENDIF
+ EXEC SQL
+        SET :SAN-NUM = IDENTITY_VAL_LOCAL() 
+ END-EXEC 
+ 
+ MOVE  SAN-NUM TO PY-RES-INT
+ MOVE 25 TO MAN-BOOKER
+ MOVE 20 TO AGE
+ DISPLAY AGE
+ EXIT.
+
+
+Answer:
+```python
+va_tab = 'HELLO WORLD'
+ca_payment = '45'
+
+if man_booker == 24:
+    try:
+        with sqlite3.connect('example.db') as conn:
+            cursor = conn.cursor()
+            cursor.execute("INSERT INTO ADDR (NUMBER, NAME) VALUES (?, ?)", (san_num, shap_value))
+            conn.commit()
+    except sqlite3.Error as e:
+        print(f"Error: {e}")
+        perform_tran_rate_interval()
+    finally:
+        cursor.close()
+
+try:
+    with sqlite3.connect('example.db') as conn:
+        cursor = conn.cursor()
+        cursor.execute("SET ?", (san_num,))
+        conn.commit()
+except sqlite3.Error as e:
+    print(f"Error: {e}")
+finally:
+    cursor.close()
+
+py_res_int = san_num
+man_booker = 25
+age = 20
+print(age)
+exit()
+
+```