diff options
Diffstat (limited to 'test/test_prime.rb')
| -rw-r--r-- | test/test_prime.rb | 17 |
1 files changed, 7 insertions, 10 deletions
diff --git a/test/test_prime.rb b/test/test_prime.rb index b48ccae319..7f15894abf 100644 --- a/test/test_prime.rb +++ b/test/test_prime.rb @@ -140,17 +140,14 @@ class TestPrime < Test::Unit::TestCase end def test_prime? - # zero and unit - assert_not_predicate(0, :prime?) - assert_not_predicate(1, :prime?) - - # small primes - assert_predicate(2, :prime?) - assert_predicate(3, :prime?) + PRIMES.each do |p| + assert_predicate(p, :prime?) + end - # squared prime - assert_not_predicate(4, :prime?) - assert_not_predicate(9, :prime?) + composites = (0..PRIMES.last).to_a - PRIMES + composites.each do |c| + assert_not_predicate(c, :prime?) + end # mersenne numbers assert_predicate((2**31-1), :prime?) |