Fix numpy arguments

equinor · Sep 10, 2024 · e5e5980 · e5e5980
1 parent 423c098
commit e5e5980
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_realization.py b/tests/test_realization.py
@@ -328,7 +328,7 @@ def test_volumetric_rates():
 
     # Pick 10 **random** dates to get the volumetric rates between:
     daily_dates = real.get_smry_dates(freq="daily", normalize=False)
-    rng = np.random.Generator()
+    rng = np.random.default_rng()
     subset_dates = rng.choice(daily_dates, size=10, replace=False)
     subset_dates.sort()
     dcum = real.get_smry(column_keys="FOPT", time_index=subset_dates)

diff --git a/tests/test_virtualensemble.py b/tests/test_virtualensemble.py
@@ -312,7 +312,7 @@ def test_todisk(tmpdir):
         assert set(vens.keys()) == set(fromcsvdisk2.keys())
 
     # Test manual intervention:
-    rng = np.random.Generator()
+    rng = np.random.default_rng()
     fooframe = pd.DataFrame(
         data=rng.standard_normal(size=(3, 3)), columns=["FOO", "BAR", "COM"]
     )