cool, tests pass

Author: molpopgen

python/test_json_metadata.py

@@ -1,4 +1,5 @@
 import tskit
+import numpy as np
 
 
 def setup_ts():
@@ -13,9 +14,27 @@ def setup_ts():
                            "phenotypes": {"type": "array"}},
             "additionalProperties": False,
         })
+    tables.mutations.metadata_schema = tskit.metadata.MetadataSchema(
+        {
+            "codec": "json",
+            "type": "object",
+            "name": "Individual metadata",
+            "properties": {"effect_size": {"type": "number"},
+                           "dominance": {"type": "number"}},
+            "additionalProperties": False,
+        })
     return tables.tree_sequence()
 
 
 def test_individual_metadata():
     ts = setup_ts()
-    raise NotImplementedError()
+    md = ts.individual(0).metadata
+    assert md["name"] == "Jerome"
+    assert md["phenotypes"] == [0, 1, 2, 0]
+
+
+def test_mutation_metadata():
+    ts = setup_ts()
+    md = ts.mutation(0).metadata
+    assert np.isclose(md["effect_size"], -1e-3)
+    assert np.isclose(md["dominance"], 0.1)