set array storage during IntegerType serialization

asdf/_tests/tags/core/tests/test_integer.py

@@ -58,26 +58,6 @@ def test_integer_storage(tmpdir, inline):
         assert rf.tree["integer"]["string"] == str(value)
 
 
-def test_integer_storage_duplication(tmpdir):
-    tmpfile = str(tmpdir.join("integer.asdf"))
-
-    random.seed(0)
-    value = random.getrandbits(1000)
-    tree = {"integer1": IntegerType(value), "integer2": IntegerType(value)}
-
-    with asdf.AsdfFile(tree) as af:
-        af.write_to(tmpfile)
-
-    with asdf.open(tmpfile, _force_raw_types=True) as rf:
-        assert len(rf._blocks.blocks) == 1
-        assert rf.tree["integer1"]["words"]["source"] == 0
-        assert rf.tree["integer2"]["words"]["source"] == 0
-
-    with asdf.open(tmpfile) as aa:
-        assert aa.tree["integer1"] == value
-        assert aa.tree["integer2"] == value
-
-
 def test_integer_conversion():
     random.seed(0)
     value = random.getrandbits(1000)

asdf/core/_converters/integer.py

@@ -24,8 +24,7 @@ def to_yaml_tree(self, obj, tag, ctx):
         array = np.array(words, dtype=np.uint32)
 
         tree = {}
-        # TODO: No support for this yet in SerializationContext
-        # ctx.set_array_storage(array, node._storage)
+        ctx._blocks._set_array_storage(array, obj._storage)
         tree["words"] = array
         tree["sign"] = obj._sign
         tree["string"] = str(int(obj._value))