t/20-return.t

   1 #!perl -T
   2
   3 use strict;
   4 use warnings;
   5
   6 use Test::More tests => 79;
   7
   8 use Sub::Nary;
   9
  10 my $sn = Sub::Nary->new();
  11
  12 my ($x, $y, @a, %h);
  13
  14 my $exp_2 = { 1 => 0.5, 2 => 0.5 };
  15
  16 my @tests = (
  17  [ sub { return },               0 ],
  18  [ sub { return () },            0 ],
  19  [ sub { return return },        0 ],
  20  [ sub { return do { return } }, 0 ],
  21
  22  [ sub { return 1 },                           1 ],
  23  [ sub { return 1, 2 },                        2 ],
  24  [ sub { my $x = 1; $x = 2; return 3, 4, 5; }, 3 ],
  25  [ sub { do { 1; return 2, 3 } },              2 ],
  26  [ sub { do { 1; return 2, 3; 4 } },           2 ],
  27  [ sub { do { 1; return 2, return 3 } },       1 ],
  28  [ sub { do { return if $x; 2, 3 }, 4 },       { 0 => 0.5, 3 => 0.5 } ],
  29  [ sub { do { do { return if $x }, 3 }, 4 },   { 0 => 0.5, 3 => 0.5 } ],
  30  [ sub { do { return if $x; 2, 3 }, do { return 1 if $y; 4, 5, 6 } },
  31                                            { 0 => 0.5, 1 => 0.25, 5 => 0.25 } ],
  32
  33  [ sub { return $x },     1 ],
  34  [ sub { return $x, $y }, 2 ],
  35
  36  [ sub { return @a },                'list' ],
  37  [ sub { return $a[0] },             1 ],
  38  [ sub { return @a[1, 2] },          2 ],
  39  [ sub { return @a[2 .. 4] },        3 ],
  40  [ sub { return @a[1, 4 .. 7, 2] },  6 ],
  41  [ sub { return @a[do{ 1 .. 5 }] },  5 ],
  42  [ sub { return @a[do{ 1 .. $x }] }, 'list' ],
  43
  44  [ sub { return %h },              'list' ],
  45  [ sub { return $h{a} },           1 ],
  46  [ sub { return @h{qw/a b/} },     2 ],
  47  [ sub { return @h{@a[1 .. 3]} },  3 ],
  48  [ sub { return @h{@a[$y .. 3]} }, 'list' ],
  49  [ sub { return keys %h },         'list' ],
  50  [ sub { return values %h },       'list' ],
  51
  52  [ sub { return $x, $a[3], $h{c} }, 3 ],
  53  [ sub { return $x, @a },           'list' ],
  54  [ sub { return %h, $y },           'list' ],
  55
  56  [ sub { return 2 .. 4 },                  3 ],
  57  [ sub { return $x .. 3 },                 'list' ],
  58  [ sub { return 1 .. $x },                 'list' ],
  59  [ sub { return '2foo' .. 4 },             3 ],
  60  [ sub { my @a = (7, 8); return @a .. 4 }, 'list' ],
  61  [ sub { return do { return 1, 2 } .. 3 }, 2 ],
  62  [ sub { return 1 .. do { return 2, 3 } }, 2 ],
  63  [ sub { my @a = return 6, $x },           2 ],
  64
  65  [ sub { for (1, 2, 3) { return } },                                     0 ],
  66  [ sub { for (1, 2, 3) { } return 1, 2; },                               2 ],
  67  [ sub { for (do { return 1, 2, 3 }) { } return 1, 2; },                 3 ],
  68  [ sub { for ($x, 1, $y) { return 1, 2 } },                              2 ],
  69  [ sub { for (@a) { return 1, do { $x } } },                             2 ],
  70  [ sub { for (keys %h) { return do { 1 }, do { return @a[0, 2] } } },    2 ],
  71  [ sub { for my $i (1 .. 4) { return @h{qw/a b/} } },                    2 ],
  72  [ sub { for (my $i; $i < 10; ++$i) { return 1, @a[do{return 2, 3}] } }, 2 ],
  73  [ sub { return 1, 2 for 1 .. 4 },                                       2 ],
  74
  75  [ sub { while (1) { return } },            0 ],
  76  [ sub { while (1) { } return 1, 2 },       2 ],
  77  [ sub { while (1) { return 1, 2 } },       2 ],
  78  [ sub { while (1) { last; return 1, 2 } }, 2 ],
  79  [ sub { return 1, 2 while 1 },             2 ],
  80  [ sub { while (do { return 2, 3 }) { } },  2 ],
  81
  82  [ sub { eval { return } },                         0 ],
  83  [ sub { eval { return 1, 2 } },                    2 ],
  84  [ sub { eval { }; return $x, 2 },                  2 ],
  85  [ sub { return eval { 1, $x }; },                  2 ],
  86  [ sub { return 1, eval { $x, eval { $h{foo} } } }, 3 ],
  87  [ sub { return eval { return $x, eval { $y } } },  2 ],
  88  [ sub { return eval { do { eval { @a } } } },      'list' ],
  89
  90  [ sub { eval 'return 1, 2' }, 'list' ],
  91
  92  [ sub { grep { return 2, 4 } 5 .. 10 },                  2 ],
  93  [ sub { grep { $_ > 1 } do { return 2, 4; 5 .. 10 } },   2 ],
  94  [ sub { grep { return 2, 4 } () },                       0 ],
  95  [ sub { grep { return $_ ? 2 : (3, 4) } 7 .. 8 },        $exp_2 ],
  96  [ sub { grep { return 2 if $_; 3 } 7 .. 8 },
  97                                                   { 1 => 0.75, list => 0.25 } ],
  98  [ sub { grep { $_ > 1 } do { return $x ? 7 : (8, 9) } }, $exp_2 ],
  99  [ sub { grep { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; } },
 100                                            { 3 => 0.5, 1 => 0.25, 2 => 0.25 } ],
 101  [ sub { grep { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; () } },
 102                                                        { 3 => 0.5, 0 => 0.5 } ],
 103
 104  [ sub { map { return 2, 4 } 5 .. 10 },                  2 ],
 105  [ sub { map { $_ + 1 } do { return 2, 4; 5 .. 10 } },   2 ],
 106  [ sub { map { return 2, 4 } () },                       0 ],
 107  [ sub { map { return $_ ? 2 : (3, 4) } 7 .. 8 },        $exp_2 ],
 108  [ sub { map { return 2 if $_; 3 } 7 .. 8 },
 109                                                      { 1 => 0.75, 2 => 0.25 } ],
 110  [ sub { map { $_ > 1 } do { return $x ? 7 : (8, 9) } }, $exp_2 ],
 111  [ sub { map { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; } },
 112                                            { 3 => 0.5, 1 => 0.25, 2 => 0.25 } ],
 113  [ sub { map { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; () } },
 114                                                        { 3 => 0.5, 0 => 0.5 } ],
 115 );
 116
 117 my $i = 1;
 118 for (@tests) {
 119  my $r = $sn->nary($_->[0]);
 120  my $exp = ref $_->[1] ? $_->[1] : { $_->[1] => 1 };
 121  is_deeply($r, $exp, 'return test ' . $i);
 122  ++$i;
 123 }