NEW: peds-simulation action

q2_fmt/__init__.py

@@ -7,11 +7,14 @@
 # ----------------------------------------------------------------------------
 
 from ._version import get_versions
+
 from ._engraftment import cc, group_timepoints
-from ._peds import sample_peds, feature_peds, peds, peds_heatmap
+from ._peds import (sample_peds, feature_peds, peds, peds_heatmap,
+                    peds_simulation)
 
 __version__ = get_versions()['version']
 del get_versions
 
+
 __all__ = ['cc', 'sample_peds', 'feature_peds',
-           'peds', 'peds_heatmap', 'group_timepoints']
+           'peds', 'peds_heatmap', 'group_timepoints', 'peds_simulation']

q2_fmt/_examples.py

@@ -234,3 +234,27 @@ def peds_heatmap(use):
     )
 
     peds_heatmap.assert_output_type('Visualization')
+
+
+def simulation_peds_method(use):
+    md = use.init_metadata('md', peds_md_factory)
+    peds_table = use.init_artifact('peds_table', peds_feature_table_factory)
+
+    peds_stats, global_stats = use.action(
+        use.UsageAction('fmt', 'peds_simulation'),
+        use.UsageInputs(
+            table=peds_table,
+            metadata=md,
+            time_column='time_point',
+            reference_column='Donor',
+            subject_column='SubjectID'
+        ),
+        use.UsageOutputNames(
+            per_subject_stats='per_subject_stats',
+            global_stats='global_stats'
+        )
+
+    )
+
+    peds_stats.assert_output_type("StatsTable[Pairwise]")
+    global_stats.assert_output_type("StatsTable[Pairwise]")