Add assert to test reset_ctx

Signed-off-by: Serge Panev <[email protected]>
apache · Sep 17, 2019 · 17051dd · 17051dd
1 parent abc5b91
commit 17051dd
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
@@ -142,6 +142,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)