test/array_test.rb in rumonade-0.4.1 vs test/array_test.rb in rumonade-0.4.2

@@ -22,14 +22,15 @@
     end
   end
 
   def test_flat_map_behaves_correctly
     assert_equal ["FOO", "BAR"], ["foo", "bar"].flat_map { |s| [s.upcase] }
-    assert_equal [2, 4, 6], [1, 2, 3].flat_map { |i| i * 2 }
+    assert_equal [2, 4, 6], [1, 2, 3].flat_map { |i| [i * 2] }
   end
 
   def test_map_behaves_correctly
     assert_equal ["FOO", "BAR"], ["foo", "bar"].map { |s| s.upcase }
+    assert_equal [2, 4, 6], [1, 2, 3].map { |i| i * 2 }
   end
 
   def test_shallow_flatten_behaves_correctly
     assert_equal [0, 1, [2], [[3]], [[[4]]]], [0, [1], [[2]], [[[3]]], [[[[4]]]]].shallow_flatten
     assert_equal [1], [None, Some(1)].shallow_flatten
\ No newline at end of file