Implement Enum.to_h (#95)

joeldrapper · Jun 28, 2024 · a4a7cf6 · a4a7cf6
1 parent 3927f33
commit a4a7cf6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/literal/enum.rb b/lib/literal/enum.rb
@@ -132,6 +132,10 @@ def fetch(...)
 		def to_proc
 			method(:cast).to_proc
 		end
+
+		def to_h(*args)
+			@values.dup
+		end
 	end
 
 	def initialize(name, value, &block)

diff --git a/test/enum.test.rb b/test/enum.test.rb
@@ -41,6 +41,8 @@ def toggle = On
 	expect(Color[1]) == Color::Red
 	expect(Color.cast(1)) == Color::Red
 	expect(Color.to_set) == Set[Color::Red, Color::Green, Color::Blue]
+	expect(Color.to_h) == {1 => Color::Red, 2 => Color::Green, 3 => Color::Blue}
+	expect(Color.to_a) == [Color::Red, Color::Green, Color::Blue]
 	expect(Color.values) == [1, 2, 3]
 	expect([3, 2, 1].map(&Color)) == [Color::Blue, Color::Green, Color::Red]