use strict;
use warnings;
-use Test::More tests => 51;
+use Test::More tests => 72;
use Sub::Nary;
my ($x, $y, @a, %h);
+my $exp_2 = { 1 => 0.5, 2 => 0.5 };
+
my @tests = (
[ sub { return }, 0 ],
[ sub { return () }, 0 ],
[ sub { return $a[0] }, 1 ],
[ sub { return @a[1, 2] }, 2 ],
[ sub { return @a[2 .. 4] }, 3 ],
+ [ sub { return @a[1, 4 .. 7, 2] }, 6 ],
[ sub { return @a[do{ 1 .. 5 }] }, 5 ],
[ sub { return @a[do{ 1 .. $x }] }, 'list' ],
[ sub { return @h{qw/a b/} }, 2 ],
[ sub { return @h{@a[1 .. 3]} }, 3 ],
[ sub { return @h{@a[$y .. 3]} }, 'list' ],
+ [ sub { return keys %h }, 'list' ],
+ [ sub { return values %h }, 'list' ],
[ sub { return $x, $a[3], $h{c} }, 3 ],
[ sub { return $x, @a }, 'list' ],
[ sub { return %h, $y }, 'list' ],
- [ sub { return 1 .. 3 }, 'list' ],
+ [ sub { return 2 .. 4 }, 3 ],
+ [ sub { return $x .. 3 }, 'list' ],
+ [ sub { return 1 .. $x }, 'list' ],
+ [ sub { return '2foo' .. 4 }, 3 ],
+ [ sub { my @a = (7, 8); return @a .. 4 }, 'list' ],
+ [ sub { return do { return 1, 2 } .. 3 }, 2 ],
+ [ sub { return 1 .. do { return 2, 3 } }, 2 ],
+ [ sub { my @a = return 6, $x }, 2 ],
[ sub { for (1, 2, 3) { return } }, 0 ],
[ sub { for (1, 2, 3) { } return 1, 2; }, 2 ],
[ sub { eval 'return 1, 2' }, 'list' ],
- [ sub { return <$x> }, 'list' ],
-
- [ sub { return caller 0 }, sub { my @a = caller 0; scalar @a }->() ],
- [ sub { return localtime }, do { my @a = localtime; scalar @a } ],
+ [ sub { grep { return 2, 4 } 5 .. 10 }, 2 ],
+ [ sub { grep { $_ > 1 } do { return 2, 4; 5 .. 10 } }, 2 ],
+ [ sub { grep { return 2, 4 } () }, 0 ],
+ [ sub { grep { return $_ ? 2 : (3, 4) } 7 .. 8 }, $exp_2 ],
+ [ sub { grep { $_ > 1 } do { return $x ? 7 : (8, 9) } }, $exp_2 ],
+ [ sub { grep { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; } },
+ { 3 => 0.5, 1 => 0.25, 2 => 0.25 } ],
+ [ sub { grep { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; () } },
+ { 3 => 0.5, 0 => 0.5 } ],
+
+ [ sub { map { return 2, 4 } 5 .. 10 }, 2 ],
+ [ sub { map { $_ + 1 } do { return 2, 4; 5 .. 10 } }, 2 ],
+ [ sub { map { return 2, 4 } () }, 0 ],
+ [ sub { map { return $_ ? 2 : (3, 4) } 7 .. 8 }, $exp_2 ],
+ [ sub { map { $_ > 1 } do { return $x ? 7 : (8, 9) } }, $exp_2 ],
+ [ sub { map { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; } },
+ { 3 => 0.5, 1 => 0.25, 2 => 0.25 } ],
+ [ sub { map { return $_ ? 2 : (3, 4) } do { return 3 .. 5 if $x; () } },
+ { 3 => 0.5, 0 => 0.5 } ],
);
my $i = 1;
for (@tests) {
my $r = $sn->nary($_->[0]);
- is_deeply($r, { $_->[1] => 1 }, 'return test ' . $i);
+ my $exp = ref $_->[1] ? $_->[1] : { $_->[1] => 1 };
+ is_deeply($r, $exp, 'return test ' . $i);
++$i;
}