allow shortcut PDL::GSL::RNG->new(..)->set_seed(..)

PDLPorters · Apr 23, 2016 · bf9dd9c · bf9dd9c
1 parent 1477036
commit bf9dd9c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/Lib/GSL/RNG/gsl_random.pd b/Lib/GSL/RNG/gsl_random.pd
@@ -76,6 +76,8 @@ Usage:
 =for usage
 
    $rng->set_seed($integer);
+   # or
+   $rng = PDL::GSL::RNG->new('taus')->set_seed($integer);
 
 Example:
 
@@ -1816,8 +1818,9 @@ void
 set_seed(rng, seed)
   gsl_rng * rng
   int seed
- CODE:
+ PPCODE:
   gsl_rng_set(rng,seed);
+  XPUSHs(ST(0)); /* return self */
 
 unsigned int
 min(rng)

diff --git a/t/gsl_rng.t b/t/gsl_rng.t
@@ -16,7 +16,7 @@ BEGIN
    if ( $PDL::Config{WITH_GSL} ) {
       eval " use PDL::GSL::RNG; ";
       unless ($@) {
-         plan tests => 18;
+         plan tests => 19;
       } else {
          plan skip_all => "PDL::GSL::RNG not installed.";
       }
@@ -40,6 +40,9 @@ $rng->set_seed(666);
 
 ok(1,'set_seed(); function');
 
+my $rng2 = PDL::GSL::RNG->new('taus')->set_seed(666);
+is(ref $rng2, 'PDL::GSL::RNG', 'PDL::GSL::RNG->new(..)->set_seed(..)');
+
 # min() function Test: 
 $min = $rng->min(); $max = $rng->max();