git-svn-id: https://svn.apache.org/repos/asf/hama/trunk@1679362 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/core/src/main/java/org/apache/hama/bsp/BSPJobClient.java b/core/src/main/java/org/apache/hama/bsp/BSPJobClient.java
index cd45d95..4600fd1 100644
--- a/core/src/main/java/org/apache/hama/bsp/BSPJobClient.java
+++ b/core/src/main/java/org/apache/hama/bsp/BSPJobClient.java
@@ -584,7 +584,8 @@
             && job.getConfiguration().get(Constants.RUNTIME_PARTITIONING_CLASS) != null
             && job.get("bsp.partitioning.runner.job") == null) {
           String[] extractPartitionID = ((FileSplit) split).getPath().getName().split("[-]");
-          rawSplit.setPartitionID(Integer.parseInt(extractPartitionID[1]));
+          if(extractPartitionID.length > 1)
+            rawSplit.setPartitionID(Integer.parseInt(extractPartitionID[1]));
         }
 
         rawSplit.setClassName(split.getClass().getName());
diff --git a/core/src/test/java/org/apache/hama/bsp/TestKeyValueTextInputFormat.java b/core/src/test/java/org/apache/hama/bsp/TestKeyValueTextInputFormat.java
index 4b548fd..c2486cd 100644
--- a/core/src/test/java/org/apache/hama/bsp/TestKeyValueTextInputFormat.java
+++ b/core/src/test/java/org/apache/hama/bsp/TestKeyValueTextInputFormat.java
@@ -78,6 +78,7 @@
 
         int expectedPeerId = Math.abs(key.hashCode() % numTasks);
 
+        System.out.println(peer.getPeerName() + ", " + key + ", " + value + ", " + expectedPeerId);
         if (expectedPeerId == peer.getPeerIndex()) {
           expectedKeys.put(new Text(key), new Text(value));
         } else {