Literal::Array#take

Closes #207
joeldrapper · Dec 2, 2024 · f1354df · f1354df
1 parent 11f94d1
commit f1354df
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/literal/array.rb b/lib/literal/array.rb
@@ -484,6 +484,10 @@ def sort!(...)
 		self
 	end
 
+	def take(...)
+		__with__(@__value__.take(...))
+	end
+
 	def to_a
 		@__value__.dup
 	end

diff --git a/test/array.test.rb b/test/array.test.rb
@@ -599,3 +599,12 @@
 	expect(result.to_a) == [2, 3, 1]
 	expect(array.to_a) == [1, 2, 3]
 end
+
+test "#take takes the first n elements" do
+	array = Literal::Array(Integer).new(1, 2, 3, 4, 5)
+
+	result = array.take(2)
+	assert Literal::Array(Integer) === result
+	expect(result.to_a) == [1, 2]
+	expect(array.to_a) == [1, 2, 3, 4, 5]
+end