Add tests

Author: devmotion

.travis.yml

@@ -12,17 +12,24 @@ branches:
 os:
   - linux
   - osx
-julia:
-  - 1.0
-  - 1
-  - nightly
+matrix:
+  include:
+    - julia: 1.0
+    - julia: 1
+      env: JULIA_NUM_THREADS=1
+    - julia: 1
+      env: JULIA_NUM_THREADS=2
+    - julia: nightly
+      env: JULIA_NUM_THREADS=1
+    - julia: nightly
+      env: JULIA_NUM_THREADS=2
 matrix:
   allow_failures:
     - julia: nightly
   fast_finish: true
 notifications:
   email: false
 after_success:
-  - if [[ $TRAVIS_JULIA_VERSION = 1 ]] && [[ $TRAVIS_OS_NAME = linux ]]; then
+  - if [[ $TRAVIS_JULIA_VERSION = 1 ]] && [[ $JULIA_NUM_THREADS = 1 ]] && [[ $TRAVIS_OS_NAME = linux ]]; then
       julia -e 'using Pkg; Pkg.add("Coverage"); using Coverage; Codecov.submit(process_folder())';
     fi

test/compiler.jl

@@ -215,6 +215,7 @@ end
             global sampler_ = _sampler
             global model_ = _model
             global context_ = _context
+            global rng_ = _rng
             global lp = getlogp(_varinfo)
             return x
         end
@@ -226,6 +227,7 @@ end
         @test model_ === model
         @test sampler_ === SampleFromPrior()
         @test context_ === DefaultContext()
+        @test rng_ isa Random.AbstractRNG
 
         # disable warnings
         @model testmodel(x) = begin
@@ -234,6 +236,7 @@ end
             global sampler_ = _sampler
             global model_ = _model
             global context_ = _context
+            global rng_ = _rng
             global lp = getlogp(_varinfo)
             return x
         end false

test/model.jl

@@ -26,4 +26,22 @@ Random.seed!(1234)
         @test ljoint ≈ lprior + llikelihood
         @test ljoint ≈ lp
     end
+
+    @testset "rng" begin
+        model = gdemo_default
+
+        for sampler in (SampleFromPrior(), SampleFromUniform())
+            for i in 1:10
+                Random.seed!(100 + i)
+                vi = VarInfo()
+                model(Random.GLOBAL_RNG, vi, sampler)
+                vals = DynamicPPL.getall(vi)
+
+                Random.seed!(100 + i)
+                vi = VarInfo()
+                model(Random.GLOBAL_RNG, vi, sampler)
+                @test DynamicPPL.getall(vi) == vals
+            end
+        end
+    end
 end