1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/test/ruby/test_pattern_matching.rb
2019-09-26 15:10:48 +09:00

1192 lines
18 KiB
Ruby

# frozen_string_literal: true
require 'test/unit'
verbose, $VERBOSE = $VERBOSE, nil # suppress "warning: Pattern matching is experimental, and the behavior may change in future versions of Ruby!"
eval "\n#{<<~'END_of_GUARD'}", binding, __FILE__, __LINE__
class TestPatternMatching < Test::Unit::TestCase
class C
class << self
attr_accessor :keys
end
def initialize(obj)
@obj = obj
end
def deconstruct
@obj
end
def deconstruct_keys(keys)
C.keys = keys
@obj
end
end
def test_basic
assert_block do
case 0
in 0
true
else
false
end
end
assert_block do
case 0
in 1
false
else
true
end
end
assert_raise(NoMatchingPatternError) do
case 0
in 1
false
end
end
begin
o = [0]
case o
in 1
false
end
rescue => e
assert_match o.inspect, e.message
end
assert_block do
begin
true
ensure
case 0
in 0
false
end
end
end
assert_block do
begin
true
ensure
case 0
in 1
else
false
end
end
end
assert_raise(NoMatchingPatternError) do
begin
ensure
case 0
in 1
end
end
end
assert_block do
verbose, $VERBOSE = $VERBOSE, nil # suppress "warning: Pattern matching is experimental, and the behavior may change in future versions of Ruby!"
eval(%q{
case true
in a
a
end
})
ensure
$VERBOSE = verbose
end
assert_block do
tap do |a|
tap do
case true
in a
a
end
end
end
end
assert_raise(NoMatchingPatternError) do
o = BasicObject.new
def o.match
case 0
in 1
end
end
o.match
end
end
def test_modifier
assert_block do
case 0
in a if a == 0
true
end
end
assert_block do
case 0
in a if a != 0
else
true
end
end
assert_block do
case 0
in a unless a != 0
true
end
end
assert_block do
case 0
in a unless a == 0
else
true
end
end
end
def test_as_pattern
assert_block do
case 0
in 0 => a
a == 0
end
end
end
def test_alternative_pattern
assert_block do
[0, 1].all? do |i|
case i
in 0 | 1
true
end
end
end
assert_block do
case 0
in _ | _a
true
end
end
assert_syntax_error(%q{
case 0
in a | 0
end
}, /illegal variable in alternative pattern/)
end
def test_var_pattern
# NODE_DASGN_CURR
assert_block do
case [0, 1]
in a, a
a == 1
end
end
# NODE_DASGN
b = 0
assert_block do
case [0, 1]
in b, b
b == 1
end
end
# NODE_LASGN
case [0, 1]
in c, c
assert_equal(1, c)
else
flunk
end
assert_syntax_error(%q{
case 0
in ^a
end
}, /no such local variable/)
end
def test_literal_value_pattern
assert_block do
case [nil, self, true, false]
in [nil, self, true, false]
true
end
end
assert_block do
case [0d170, 0D170, 0xaa, 0xAa, 0xAA, 0Xaa, 0XAa, 0XaA, 0252, 0o252, 0O252]
in [0d170, 0D170, 0xaa, 0xAa, 0xAA, 0Xaa, 0XAa, 0XaA, 0252, 0o252, 0O252]
true
end
case [0b10101010, 0B10101010, 12r, 12.3r, 1i, 12.3ri]
in [0b10101010, 0B10101010, 12r, 12.3r, 1i, 12.3ri]
true
end
end
assert_block do
x = 'x'
case ['a', 'a', x]
in ['a', "a", "#{x}"]
true
end
end
assert_block do
case ["a\n"]
in [<<END]
a
END
true
end
end
assert_block do
case [:a, :"a"]
in [:a, :"a"]
true
end
end
assert_block do
case [0, 1, 2, 3, 4, 5]
in [0..1, 0...2, 0.., 0..., (...5), (..5)]
true
end
end
assert_syntax_error(%q{
case 0
in a..b
end
}, /unexpected/)
assert_block do
case 'abc'
in /a/
true
end
end
assert_block do
case 0
in ->(i) { i == 0 }
true
end
end
assert_block do
case [%(a), %q(a), %Q(a), %w(a), %W(a), %i(a), %I(a), %s(a), %x(echo a), %(), %q(), %Q(), %w(), %W(), %i(), %I(), %s(), 'a']
in [%(a), %q(a), %Q(a), %w(a), %W(a), %i(a), %I(a), %s(a), %x(echo a), %(), %q(), %Q(), %w(), %W(), %i(), %I(), %s(), %r(a)]
true
end
end
assert_block do
case [__FILE__, __LINE__ + 1, __ENCODING__]
in [__FILE__, __LINE__, __ENCODING__]
true
end
end
end
def test_constant_value_pattern
assert_block do
case 0
in Integer
true
end
end
assert_block do
case 0
in Object::Integer
true
end
end
assert_block do
case 0
in ::Object::Integer
true
end
end
end
def test_pin_operator_value_pattern
assert_block do
a = /a/
case 'abc'
in ^a
true
end
end
assert_block do
case [0, 0]
in a, ^a
a == 0
end
end
end
def test_array_pattern
assert_block do
[[0], C.new([0])].all? do |i|
case i
in 0,;
true
end
end
end
assert_block do
[[0, 1], C.new([0, 1])].all? do |i|
case i
in 0,;
true
end
end
end
assert_block do
[[], C.new([])].all? do |i|
case i
in 0,;
else
true
end
end
end
assert_block do
[[0, 1], C.new([0, 1])].all? do |i|
case i
in 0, 1
true
end
end
end
assert_block do
[[0], C.new([0])].all? do |i|
case i
in 0, 1
else
true
end
end
end
assert_block do
[[], C.new([])].all? do |i|
case i
in *a
a == []
end
end
end
assert_block do
[[0], C.new([0])].all? do |i|
case i
in *a
a == [0]
end
end
end
assert_block do
[[0], C.new([0])].all? do |i|
case i
in *a, 0, 1
else
true
end
end
end
assert_block do
[[0, 1], C.new([0, 1])].all? do |i|
case i
in *a, 0, 1
a == []
end
end
end
assert_block do
[[0, 1, 2], C.new([0, 1, 2])].all? do |i|
case i
in *a, 1, 2
a == [0]
end
end
end
assert_block do
[[], C.new([])].all? do |i|
case i
in *;
true
end
end
end
assert_block do
[[0], C.new([0])].all? do |i|
case i
in *, 0, 1
else
true
end
end
end
assert_block do
[[0, 1], C.new([0, 1])].all? do |i|
case i
in *, 0, 1
true
end
end
end
assert_block do
[[0, 1, 2], C.new([0, 1, 2])].all? do |i|
case i
in *, 1, 2
true
end
end
end
assert_block do
case C.new([0])
in C(0)
true
end
end
assert_block do
case C.new([0])
in Array(0)
else
true
end
end
assert_block do
case C.new([])
in C()
true
end
end
assert_block do
case C.new([])
in Array()
else
true
end
end
assert_block do
case C.new([0])
in C[0]
true
end
end
assert_block do
case C.new([0])
in Array[0]
else
true
end
end
assert_block do
case C.new([])
in C[]
true
end
end
assert_block do
case C.new([])
in Array[]
else
true
end
end
assert_block do
case []
in []
true
end
end
assert_block do
case C.new([])
in []
true
end
end
assert_block do
case [0]
in [0]
true
end
end
assert_block do
case C.new([0])
in [0]
true
end
end
assert_block do
case [0]
in [0,]
true
end
end
assert_block do
case [0, 1]
in [0,]
true
end
end
assert_block do
case []
in [0, *a]
else
true
end
end
assert_block do
case [0]
in [0, *a]
a == []
end
end
assert_block do
case [0]
in [0, *a, 1]
else
true
end
end
assert_block do
case [0, 1]
in [0, *a, 1]
a == []
end
end
assert_block do
case [0, 1, 2]
in [0, *a, 2]
a == [1]
end
end
assert_block do
case []
in [0, *]
else
true
end
end
assert_block do
case [0]
in [0, *]
true
end
end
assert_block do
case [0, 1]
in [0, *]
true
end
end
assert_block do
case []
in [0, *a]
else
true
end
end
assert_block do
case [0]
in [0, *a]
a == []
end
end
assert_block do
case [0, 1]
in [0, *a]
a == [1]
end
end
assert_block do
case [0]
in [0, *, 1]
else
true
end
end
assert_block do
case [0, 1]
in [0, *, 1]
true
end
end
end
def test_hash_pattern
assert_block do
[{}, C.new({})].all? do |i|
case i
in a: 0
else
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in a: 0
true
end
end
end
assert_block do
[{a: 0, b: 1}, C.new({a: 0, b: 1})].all? do |i|
case i
in a: 0
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in a: 0, b: 1
else
true
end
end
end
assert_block do
[{a: 0, b: 1}, C.new({a: 0, b: 1})].all? do |i|
case i
in a: 0, b: 1
true
end
end
end
assert_block do
[{a: 0, b: 1, c: 2}, C.new({a: 0, b: 1, c: 2})].all? do |i|
case i
in a: 0, b: 1
true
end
end
end
assert_block do
[{}, C.new({})].all? do |i|
case i
in a:
else
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in a:
a == 0
end
end
end
assert_block do
[{a: 0, b: 1}, C.new({a: 0, b: 1})].all? do |i|
case i
in a:
a == 0
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in "a": 0
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in "a":;
a == 0
end
end
end
assert_block do
[{}, C.new({})].all? do |i|
case i
in **a
a == {}
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in **a
a == {a: 0}
end
end
end
assert_block do
[{}, C.new({})].all? do |i|
case i
in **;
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in **;
true
end
end
end
assert_block do
[{}, C.new({})].all? do |i|
case i
in a:, **b
else
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in a:, **b
a == 0 && b == {}
end
end
end
assert_block do
[{a: 0, b: 1}, C.new({a: 0, b: 1})].all? do |i|
case i
in a:, **b
a == 0 && b == {b: 1}
end
end
end
assert_block do
[{}, C.new({})].all? do |i|
case i
in **nil
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in **nil
else
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in a:, **nil
true
end
end
end
assert_block do
[{a: 0, b: 1}, C.new({a: 0, b: 1})].all? do |i|
case i
in a:, **nil
else
true
end
end
end
assert_block do
case C.new({a: 0})
in C(a: 0)
true
end
end
assert_block do
case {a: 0}
in C(a: 0)
else
true
end
end
assert_block do
case C.new({a: 0})
in C[a: 0]
true
end
end
assert_block do
case {a: 0}
in C[a: 0]
else
true
end
end
assert_block do
[{}, C.new({})].all? do |i|
case i
in {a: 0}
else
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in {a: 0}
true
end
end
end
assert_block do
[{a: 0, b: 1}, C.new({a: 0, b: 1})].all? do |i|
case i
in {a: 0}
true
end
end
end
assert_block do
[{}, C.new({})].all? do |i|
case i
in {}
true
end
end
end
assert_block do
[{a: 0}, C.new({a: 0})].all? do |i|
case i
in {}
else
true
end
end
end
assert_syntax_error(%q{
case _
in a:, a:
end
}, /duplicated key name/)
assert_syntax_error(%q{
case _
in a?:
end
}, /key must be valid as local variables/)
assert_block do
case {a?: true}
in a?: true
true
end
end
assert_syntax_error(%q{
case _
in "a-b":
end
}, /key must be valid as local variables/)
assert_block do
case {"a-b": true}
in "a-b": true
true
end
end
assert_syntax_error(%q{
case _
in "#{a}": a
end
}, /symbol literal with interpolation is not allowed/)
assert_syntax_error(%q{
case _
in "#{a}":
end
}, /symbol literal with interpolation is not allowed/)
end
def test_paren
assert_block do
case 0
in (0)
true
end
end
end
def test_invalid_syntax
assert_syntax_error(%q{
case 0
in a, b:
end
}, /unexpected/)
assert_syntax_error(%q{
case 0
in [a:]
end
}, /unexpected/)
assert_syntax_error(%q{
case 0
in {a}
end
}, /unexpected/)
assert_syntax_error(%q{
case 0
in {0 => a}
end
}, /unexpected/)
end
################################################################
class CTypeError
def deconstruct
nil
end
def deconstruct_keys(keys)
nil
end
end
def test_deconstruct
assert_raise(TypeError) do
case CTypeError.new
in []
end
end
end
def test_deconstruct_keys
assert_raise(TypeError) do
case CTypeError.new
in {}
end
end
assert_block do
case {}
in {}
C.keys == nil
end
end
assert_block do
case C.new({a: 0, b: 0, c: 0})
in {a: 0, b:}
C.keys == [:a, :b]
end
end
assert_block do
case C.new({a: 0, b: 0, c: 0})
in {a: 0, b:, **}
C.keys == [:a, :b]
end
end
assert_block do
case C.new({a: 0, b: 0, c: 0})
in {a: 0, b:, **r}
C.keys == nil
end
end
assert_block do
case C.new({a: 0, b: 0, c: 0})
in {**}
C.keys == []
end
end
assert_block do
case C.new({a: 0, b: 0, c: 0})
in {**r}
C.keys == nil
end
end
end
################################################################
class TestPatternMatchingRefinements < Test::Unit::TestCase
class C1
def deconstruct
[:C1]
end
end
class C2
end
module M
refine Array do
def deconstruct
[0]
end
end
refine Hash do
def deconstruct_keys(_)
{a: 0}
end
end
refine C2.singleton_class do
def ===(obj)
obj.kind_of?(C1)
end
end
end
using M
def test_refinements
assert_block do
case []
in [0]
true
end
end
assert_block do
case {}
in {a: 0}
true
end
end
assert_block do
case C1.new
in C2(:C1)
true
end
end
end
end
################################################################
def test_struct
assert_block do
s = Struct.new(:a, :b)
case s[0, 1]
in 0, 1
true
end
end
end
################################################################
def test_modifier_in
assert_equal true, (1 in a)
assert_equal 1, a
end
end
END_of_GUARD
$VERBOSE = verbose