refactor by meru

lib/transport.rb

@@ -1,11 +1,32 @@
 def transport(source)
-  array = source.split("\n").map {|s| s.split(" ")}
+  array = split_numbers(split_rows(source))
   rows_count = array.first.count
 
   transported_array = []
-  0.upto(rows_count - 1) do |i|
-    transported_array << array.map {|a| a[i]}
+  rows_count.times do |i|
+    transported_array << set_number(array, i)
   end
 
-  transported_array.map {|s| s.join(" ")}.join("\n")
-end
+  join_rows(join_numbers(transported_array))
+end
+
+def split_rows(string)
+  string.split("\n")
+end
+
+def split_numbers(array)
+  array.map{|a| a.split(" ")}
+end
+
+def join_rows(array)
+  array.join("\n")
+end
+
+def join_numbers(array)
+  array.map{|a| a.join(" ")}
+end
+
+def set_number(array, i)
+  array.map {|a| a[i]}
+end
+

test/transport_test.rb

@@ -11,4 +11,4 @@ def test_transport
     output = "1 4 7 10\n2 5 8 11\n3 6 9 12"
     assert_equal output, transport(input)
   end
-end
+end