adding unit test for divide

Samreay · Nov 6, 2023 · 97d277e · 97d277e
1 parent edb43b6
commit 97d277e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_chain.py b/tests/test_chain.py
@@ -97,3 +97,9 @@ def test_color_data_p1(self):
         color_data = chain.color_data
         assert color_data is not None
         assert np.allclose(self.df["a"].to_numpy(), color_data)
+
+    def test_divide(self):
+
+        n_walkers = 10
+        result = Chain(samples=self.df, name=self.n, walkers=n_walkers).divide()
+        assert len(result) == n_walkers