t/23-branch.t

   1 #!perl -T
   2
   3 use strict;
   4 use warnings;
   5
   6 use Test::More tests => 14;
   7
   8 use Sub::Nary;
   9
  10 my $sn = Sub::Nary->new();
  11
  12 my ($x, $y, @a, %h);
  13
  14 sub ret12 {
  15  if ($x) {
  16   return 1
  17  } else {
  18   return 1, 2
  19  }
  20 }
  21
  22 sub ret1l { $x ? 1 : @_ }
  23
  24 sub ret1234 {
  25  if ($x) {
  26   return 1, 2
  27  } elsif ($h{foo}) {
  28   return 3, @a[4, 5];
  29  } elsif (@a) {
  30   return @h{qw/a b c/}, $y
  31  }
  32 }
  33
  34 sub retinif {
  35  if (return 1, 2) {
  36   return 1, 2, 3
  37  } else {
  38   return @_[0 .. 3]
  39  }
  40 }
  41
  42 my @tests = (
  43  [ \&ret12,                    { 1 => 0.5, 2 => 0.5 } ],
  44  [ sub { 1, ret12 },           { 2 => 0.5, 3 => 0.5 } ],
  45  [ sub { 1, do { ret12, 3 } }, { 3 => 0.5, 4 => 0.5 } ],
  46  [ sub { @_[ret12()] },        { 1 => 0.5, 2 => 0.5 } ],
  47
  48  [ sub { ret12, ret12 },    { 2 => 0.25, 3 => 0.5, 4 => 0.25 } ],
  49  [ sub { ret12, 0, ret12 }, { 3 => 0.25, 4 => 0.5, 5 => 0.25 } ],
  50  [ sub { ret12, @a },       { list => 1 } ],
  51  [ sub { %h, ret12 },       { list => 1 } ],
  52
  53  [ sub { if ($y) { ret12 } else { ret12 } }, { 1 => 0.5, 2 => 0.5 } ],
  54
  55  [ \&ret1l,                     { 1 => 0.5, list => 0.5 } ],
  56  [ sub { $_[0], ret1l },        { 2 => 0.5, list => 0.5 } ],
  57  [ sub { ret1l, ret1l, ret1l }, { 3 => 0.125, list => 0.875 } ],
  58
  59  [ \&ret1234, { map { $_ => 0.25 } 1 .. 4 } ],
  60
  61  [ \&retinif, { 2 => 1 } ],
  62 );
  63
  64 my $i = 1;
  65 for (@tests) {
  66  my $r = $sn->nary($_->[0]);
  67  is_deeply($r, $_->[1], 'branch test ' . $i);
  68  ++$i;
  69 }
  70