Add assert to test reset_ctx

Signed-off-by: Serge Panev <[email protected]>
apache · Sep 25, 2019 · 4b817e2 · 4b817e2
1 parent 1e442ce
commit 4b817e2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -143,6 +143,9 @@ def test_parameter_dict():
     params3.initialize(ctx=ctx)
     list_contexts = [mx.cpu(42), mx.cpu(24)]
     params3.reset_ctx(list_contexts)
+    for p in params3.values():
+        assert set(p.list_ctx()) == set(list_contexts)
+
     # and test list_ctx
     assert set(params3.list_ctx()) == set(list_contexts)