Test default behavior

apache · Oct 25, 2019 · 803f765 · 803f765
1 parent 74dd12f
commit 803f765
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -1540,6 +1540,17 @@ def __init__(self, b1, b2):
     c = C(b1, b2)
     c.load_parameters('tmp.params')
 
+    # Test default behavior
+    c.save_parameters('tmp2.params', deduplicate=False)
+
+    params = mx.nd.load('tmp2.params')
+    assert len(params) == 2  # Only a single copy of the shared parameter is saved
+
+    b1 = B()
+    b2 = B(b1.collect_params())
+    c = C(b1, b2)
+    c.load_parameters('tmp2.params')
+
 @with_seed()
 def test_symbol_block_save_load():
     class Net(gluon.HybridBlock):