From de80e9ed0e2ee6d3e21d03ded2de9eadd270bad5 Mon Sep 17 00:00:00 2001
From: Doug Branton <brantd@uw.edu>
Date: Mon, 1 Jul 2024 10:20:32 -0700
Subject: [PATCH] add testing of seed stability

---
 tests/nested_dask/test_datasets.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/nested_dask/test_datasets.py b/tests/nested_dask/test_datasets.py
index 26f9ad7..7a2e66a 100644
--- a/tests/nested_dask/test_datasets.py
+++ b/tests/nested_dask/test_datasets.py
@@ -1,4 +1,5 @@
 import nested_dask as nd
+import pytest
 
 
 def test_generate_data():
@@ -18,3 +19,8 @@ def test_generate_data():
     # test the length
     assert len(generate_1) == 10
     assert len(generate_1.nested.nest.to_flat()) == 1000
+
+    # test seed stability
+    assert pytest.approx(generate_1.compute().loc[0]["a"], 0.1) == 0.417
+    assert pytest.approx(generate_1.compute().loc[0]["b"], 0.1) == 0.838
+    assert pytest.approx(generate_1.nested.nest.to_flat().compute().iloc[0]["t"], 0.1) == 16.015