test cases: is_training/is_recording

dmlc · Dec 11, 2017 · 617b18f · 617b18f
1 parent 5db7fbd
commit 617b18f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/unittest/autograd.jl b/test/unittest/autograd.jl
@@ -105,19 +105,19 @@ end  # function test_record
 function test_is_recording()
   info("AutoGrad::is_recording")
   mx.record() do
-    @test is_recording()
+    @test mx.is_recording()
   end
 end  # function test_is_recording
 
 
 function test_is_training()
   info("AutoGrad::is_training")
   mx.record() do
-    @test is_training()
+    @test mx.is_training()
   end
 
   mx.record(false) do
-    @test !is_training()
+    @test !mx.is_training()
   end
 end  # function test_is_training