Check the whole result only on different length in Array#flatten!

opal · Oct 24, 2013 · 6a62d0b · 6a62d0b
1 parent 5aa9af2
commit 6a62d0b
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/corelib/array.rb b/corelib/array.rb
@@ -622,14 +622,16 @@ def flatten!(level = undefined)
     %x{
       var flattened = #{flatten level};
 
-      for (var i = 0, length = self.length; i < length; i++) {
-        if (self[i] !== flattened[i]) {
-          break;
+      if (self.length == flattened.length) {
+        for (var i = 0, length = self.length; i < length; i++) {
+          if (self[i] !== flattened[i]) {
+            break;
+          }
         }
-      }
 
-      if (i == length) {
-        return nil;
+        if (i == length) {
+          return nil;
+        }
       }
 
       #{replace `flattened`};