t/23-branch.t

   1 #!perl -T
   2
   3 use strict;
   4 use warnings;
   5
   6 use Test::More tests => 16;
   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 ret12b {
  23  if ($x) {
  24   return 1
  25  }
  26  return 1, 2
  27 }
  28
  29 sub ret12c {
  30  if (@a) {
  31   return 1, 2
  32  }
  33 }
  34
  35 sub ret1l { $x ? 1 : @_ }
  36
  37 sub ret1234 {
  38  if ($x) {
  39   return 1, 2
  40  } elsif ($h{foo}) {
  41   return 3, @a[4, 5];
  42  } elsif (@a) {
  43   return @h{qw/a b c/}, $y
  44  }
  45 }
  46
  47 sub retinif {
  48  if (return 1, 2) {
  49   return 1, 2, 3
  50  } else {
  51   return @_[0 .. 3]
  52  }
  53 }
  54
  55 my @tests = (
  56  [ \&ret12,                    { 1 => 0.5, 2 => 0.5 } ],
  57  [ \&ret12b,                   { 1 => 0.5, 2 => 0.5 } ],
  58  [ \&ret12c,                   { 1 => 0.5, 2 => 0.5 } ],
  59  [ sub { 1, ret12 },           { 2 => 0.5, 3 => 0.5 } ],
  60  [ sub { 1, do { ret12, 3 } }, { 3 => 0.5, 4 => 0.5 } ],
  61  [ sub { @_[ret12()] },        { 1 => 0.5, 2 => 0.5 } ],
  62
  63  [ sub { ret12, ret12 },    { 2 => 0.25, 3 => 0.5, 4 => 0.25 } ],
  64  [ sub { ret12, 0, ret12 }, { 3 => 0.25, 4 => 0.5, 5 => 0.25 } ],
  65  [ sub { ret12, @a },       { list => 1 } ],
  66  [ sub { %h, ret12 },       { list => 1 } ],
  67
  68  [ sub { if ($y) { ret12 } else { ret12 } }, { 1 => 0.5, 2 => 0.5 } ],
  69
  70  [ \&ret1l,                     { 1 => 0.5, list => 0.5 } ],
  71  [ sub { $_[0], ret1l },        { 2 => 0.5, list => 0.5 } ],
  72  [ sub { ret1l, ret1l, ret1l }, { 3 => 0.125, list => 0.875 } ],
  73
  74  [ \&ret1234, { 2 => 0.5, 3 => 0.25, 4 => 0.125, 1 => 0.125 } ],
  75
  76  [ \&retinif, { 2 => 1 } ],
  77 );
  78
  79 my $i = 1;
  80 for (@tests) {
  81  my $r = $sn->nary($_->[0]);
  82  my $exp = ref $_->[1] ? $_->[1] : { $_->[1] => 1 };
  83  is_deeply($r, $exp, 'branch test ' . $i);
  84  ++$i;
  85 }
  86