1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
///
/// @file next_prime1.cpp
/// @brief Test next_prime() of primesieve::iterator.
///
/// Copyright (C) 2017 Kim Walisch, <kim.walisch@gmail.com>
///
/// This file is distributed under the BSD License. See the COPYING
/// file in the top level directory.
///
#include <primesieve.hpp>
#include <stdint.h>
#include <cstdlib>
#include <exception>
#include <iostream>
#include <vector>
using namespace std;
void check(bool OK)
{
cout << " " << (OK ? "OK" : "ERROR") << "\n";
if (!OK)
exit(1);
}
int main()
{
vector<uint64_t> primes;
primesieve::generate_primes(100000, &primes);
primesieve::iterator it;
uint64_t stop = primes.size() - 1;
uint64_t prime;
for (uint64_t i = 0; i < stop; i++)
{
it.skipto(primes[i] - 1);
prime = it.next_prime();
cout << "next_prime(" << primes[i] - 1 << ") = " << prime;
check(prime == primes[i]);
it.skipto(primes[i]);
prime = it.next_prime();
cout << "next_prime(" << primes[i] << ") = " << prime;
check(prime == primes[i + 1]);
}
it.skipto(0);
prime = it.next_prime();
uint64_t sum = 0;
// iterate over the primes below 10^9
for (; prime < 1000000000; prime = it.next_prime())
sum += prime;
cout << "Sum of the primes below 10^9 = " << sum;
check(sum == 24739512092254535ull);
it.skipto(primes.back() - 200, primes.back());
prime = it.next_prime();
while (prime <= primes.back())
prime = it.next_prime();
for (uint64_t i = 1; i < 1000; i++)
{
uint64_t old = prime;
uint64_t p = primes[primes.size() - i];
prime = it.prev_prime();
cout << "prev_prime(" << old << ") = " << prime;
check(prime == p);
}
cout << endl;
cout << "All tests passed successfully!" << endl;
return 0;
}
|
