dry up array comprehension for single block

test/fixtures/array-comprehension/if/expected.js

@@ -1,12 +0,0 @@
-var seattlers = function () {
-  var _arr = [];
-  customers.forEach(function (c) {
-    if (c.city == "Seattle") {
-      _arr.push({
-        name: c.name,
-        age: c.age
-      });
-    }
-  });
-  return _arr;
-}();

test/fixtures/array-comprehension/multiple-if/actual.js

@@ -0,0 +1 @@
+var seattlers = [for (customers of countries) for (c of customers) if (c.city == "Seattle") { name: c.name, age: c.age }];

test/fixtures/array-comprehension/multiple-if/expected.js

@@ -0,0 +1,14 @@
+var seattlers = function () {
+  var _arr = [];
+  countries.forEach(function (customers) {
+    customers.forEach(function (c) {
+      if (c.city == "Seattle") {
+        _arr.push({
+          name: c.name,
+          age: c.age
+        });
+      }
+    });
+  });
+  return _arr;
+}();

test/fixtures/array-comprehension/single-if/expected.js

@@ -0,0 +1,8 @@
+var seattlers = customers.filter(function (c) {
+  return c.city == "Seattle";
+}).map(function (c) {
+  return {
+    name: c.name,
+    age: c.age
+  };
+});

test/fixtures/array-comprehension/single/expected.js

@@ -1,9 +1,3 @@
-var arr = (function () {
-  var _arr = [];
-
-  [1, 2, 3].forEach(function (i) {
-    _arr.push(i * i);
-  });
-
-  return _arr;
-})();
+var arr = [1, 2, 3].map(function (i) {
+  return i * i;
+});