Diff of /tests/sim/test_splat.py [000000] .. [ea0fd6]

Switch to side-by-side view

--- a
+++ b/tests/sim/test_splat.py
@@ -0,0 +1,27 @@
+import unittest
+
+from inmoose.sim.splat import sim_rnaseq
+
+N = 1000
+M = 10000
+batch = (M // 2) * ["A"] + (M // 2) * ["B"]
+group = (M // 4) * ["A"] + (M // 4) * ["B"]
+group = group + group
+
+
+class Test(unittest.TestCase):
+    def test_sim_rnaseq(self):
+        """
+        check that RNA-Seq data simulation works properly
+        """
+        data = sim_rnaseq(N, M, random_state=42)
+        self.assertEqual(data.shape, (N, M))
+        data = sim_rnaseq(N, M, batch=batch)
+        data = sim_rnaseq(N, M, group=group)
+
+    def test_sim_scrnaseq(self):
+        """
+        check that scRNA-Seq data simulation works properly
+        """
+        data = sim_rnaseq(N, M, single_cell=True, random_state=51)
+        self.assertEqual(data.shape, (N, M))