flint3-sys 3.2.2

Rust bindings to the FLINT C library
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

/*
    Copyright (C) 2017 Apoorv Mishra

    This file is part of FLINT.

    FLINT is free software: you can redistribute it and/or modify it under
    the terms of the GNU Lesser General Public License (LGPL) as published
    by the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.  See <https://www.gnu.org/licenses/>.
*/

#include "test_helpers.h"
#include "ulong_extras.h"

TEST_FUNCTION_START(n_urandint, state)
{
    ulong limit, rand_num;
    slong deviation;
    int i, j, result;
    int * count;
    int * count_in_subrange;
    slong test_multiplier;

    test_multiplier = FLINT_MAX(1, flint_test_multiplier());

    /* Test for limit <= 1000 */
    count = flint_malloc(sizeof(int) * 1000);
    for (limit = 1; limit <= 1000; limit+=10)
    {
        for (i = 0; i < limit; i++)
        {
            count[i] = 0;
        }

        for (i = 0; i < 1000 * test_multiplier; i++)
        {
            rand_num = n_urandint(state, limit);
            count[rand_num]++;
        }

        result = 1;
        for (i = 0; i < limit; i++)
        {
            deviation = count[i] - (1000 * (int) test_multiplier)/limit;
            if (deviation >= WORD(100) * test_multiplier ||
                deviation <= WORD(-100) * test_multiplier)
            {
                result = 0;
                break;
            }
        }

        if (!result)
            TEST_FUNCTION_FAIL("limit = %wu, deviation = %wd\n", limit, deviation);
    }

    flint_free(count);

    /* Test for larger values of limit */
    count_in_subrange = flint_malloc(sizeof(int) * 4);
    for (i = 0; i < 1000; i+=10)
    {
        for (j = 0; j < 4; j++)
        {
            count_in_subrange[j] = 0;
        }

        limit = UWORD_MAX/(i + 2)*(i + 1);

        for (j = 0; j < 1000 * test_multiplier; j++)
        {
            rand_num = n_urandint(state, limit);

            if (rand_num >= 3*(limit >> 2))
            {
                count_in_subrange[3]++;
            }
            else if (rand_num >= 2*(limit >> 2))
            {
                count_in_subrange[2]++;
            }
            else if (rand_num >= (limit >> 2))
            {
                count_in_subrange[1]++;
            }
            else
            {
                count_in_subrange[0]++;
            }
        }

        result = 1;
        for (j = 0; j < 4; j++)
        {
            deviation = count_in_subrange[j] - ((1000 * (int) test_multiplier) >> 2);
            if (deviation >= WORD(100) * test_multiplier ||
                deviation <= WORD(-100) * test_multiplier)
            {
                result = 0;
                break;
            }
        }

        if (!result)
            TEST_FUNCTION_FAIL("limit = %wu, deviation = %wd\n", limit, deviation);
    }

    flint_free(count_in_subrange);

    TEST_FUNCTION_END(state);
}