#transpose

lib/literal/array.rb

@@ -420,6 +420,14 @@ def to_a
 
 	alias_method :to_ary, :to_a
 
+	def transpose
+		if @__type__ == Array || Literal::Array == @__type__
+			__with__(@__value__.transpose)
+		else
+			raise ArgumentError.new("#transpose expects Array type")
+		end
+	end
+
 	def uniq
 		__with__(@__value__.uniq)
 	end

test/array.test.rb

@@ -371,6 +371,25 @@
 	expect { array.replace("not an array") }.to_raise(ArgumentError)
 end
 
+test "#transpose raises if type is not Array" do
+	array = Literal::Array(Integer).new(1, 2, 3)
+
+	expect { array.transpose }.to_raise(ArgumentError)
+end
+
+test "#transpose returns a new array with rows and columns transposed" do
+	array = Literal::Array(Array).new([1, 2], [3, 4], [5, 6])
+
+	transposed = array.transpose
+	expect(transposed.to_a) == [[1, 3, 5], [2, 4, 6]]
+end
+
+test "#transpose raises if the rows are not the same length" do
+	array = Literal::Array(Array).new([1, 2], [3, 4, 5])
+
+	expect { array.transpose }.to_raise(IndexError)
+end
+
 test "#values_at returns the values at the given indexes" do
 	array = Literal::Array(Integer).new(1, 2, 3)