Skip to content

Commit

Permalink
Merge pull request #200 from bashtage/update-pract-rand-driver
Browse files Browse the repository at this point in the history
MAINT: Update practrand driver
  • Loading branch information
bashtage committed Jun 3, 2020
2 parents 7805f6a + c890735 commit 437bdf5
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 5 deletions.
5 changes: 3 additions & 2 deletions randomgen/tests/test_generator_mt19937_regressions.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,8 +59,9 @@ def test_shuffle_mixed_dimension(self):
mt19937.bit_generator.seed(12345)
shuffled = np.array(list(t), dtype=np.object)
mt19937.shuffle(shuffled)
assert_array_equal(shuffled,
np.array([t[0], t[3], t[1], t[2]], dtype=np.object))
assert_array_equal(
shuffled, np.array([t[0], t[3], t[1], t[2]], dtype=np.object)
)

def test_call_within_randomstate(self):
# Check that custom RandomState does not call into global state
Expand Down
5 changes: 3 additions & 2 deletions randomgen/tests/test_randomstate_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,9 @@ def test_shuffle_mixed_dimension(self):
random.seed(12345)
shuffled = np.array(list(t), dtype=np.object)
random.shuffle(shuffled)
assert_array_equal(shuffled,
np.array([t[0], t[3], t[1], t[2]], dtype=np.object))
assert_array_equal(
shuffled, np.array([t[0], t[3], t[1], t[2]], dtype=np.object)
)

def test_call_within_randomstate(self):
# Check that custom RandomState does not call into global state
Expand Down
3 changes: 2 additions & 1 deletion tools/practrand-driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
rg.Xoshiro256: {"output": 64, "seed": 256, "seed_size": 64},
rg.Philox: {"output": 64, "seed": 256, "seed_size": 64},
rg.SFMT: {"output": 64, "seed": 128, "seed_size": 32},
rg.LXM: {"output": 64, "seed": 128, "seed_size": 32},
}


Expand Down Expand Up @@ -129,7 +130,7 @@ def main():
formatter_class=argparse.ArgumentDefaultsHelpFormatter
)
parser.add_argument(
"-bg", "--bit_generator", type=str, default="PCG32", help="BitGenerator to use."
"-bg", "--bit_generator", type=str, default="PCG64", help="BitGenerator to use."
)
parser.add_argument(
"--load", action="store_true", help="Load BitGenerators from JSON file."
Expand Down

0 comments on commit 437bdf5

Please sign in to comment.