test_ssp_ran_discrete.c (3226B)
1 /* Copyright (C) 2015-2025 |Méso|Star> (contact@meso-star.com) 2 * 3 * This program is free software: you can redistribute it and/or modify 4 * it under the terms of the GNU General Public License as published by 5 * the Free Software Foundation, either version 3 of the License, or 6 * (at your option) any later version. 7 * 8 * This program is distributed in the hope that it will be useful, 9 * but WITHOUT ANY WARRANTY; without even the implied warranty of 10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 11 * GNU General Public License for more details. 12 * 13 * You should have received a copy of the GNU General Public License 14 * along with this program. If not, see <http://www.gnu.org/licenses/>. */ 15 16 #include "ssp.h" 17 #include "test_ssp_utils.h" 18 19 #include <string.h> 20 21 #define NSAMPS 1024 22 23 int 24 main(int argc, char** argv) 25 { 26 struct mem_allocator allocator; 27 struct ssp_ranst_discrete* ran; 28 struct ssp_rng* rng; 29 const double weights[] = { 0.5, 0.1, 0.2, 0.05, 0.15 }; 30 const size_t nweights = sizeof(weights)/sizeof(double); 31 size_t tmp[sizeof(weights)/sizeof(double)]; 32 double accum; 33 size_t i; 34 (void)argc, (void)argv; 35 36 mem_init_proxy_allocator(&allocator, &mem_default_allocator); 37 CHK(ssp_rng_create(&allocator, SSP_RNG_THREEFRY, &rng) == RES_OK); 38 39 CHK(ssp_ranst_discrete_create(NULL, NULL) == RES_BAD_ARG); 40 CHK(ssp_ranst_discrete_create(NULL, &ran) == RES_OK); 41 42 CHK(ssp_ranst_discrete_ref_get(NULL) == RES_BAD_ARG); 43 CHK(ssp_ranst_discrete_ref_get(ran) == RES_OK); 44 CHK(ssp_ranst_discrete_ref_put(NULL) == RES_BAD_ARG); 45 CHK(ssp_ranst_discrete_ref_put(ran) == RES_OK); 46 CHK(ssp_ranst_discrete_ref_put(ran) == RES_OK); 47 48 CHK(ssp_ranst_discrete_create(&allocator, &ran) == RES_OK); 49 50 CHK(ssp_ranst_discrete_setup(NULL, NULL, 0) == RES_BAD_ARG); 51 CHK(ssp_ranst_discrete_setup(ran, NULL, 0) == RES_BAD_ARG); 52 CHK(ssp_ranst_discrete_setup(NULL, weights, 0) == RES_BAD_ARG); 53 CHK(ssp_ranst_discrete_setup(ran, weights, 0) == RES_BAD_ARG); 54 CHK(ssp_ranst_discrete_setup(NULL, NULL, nweights) == RES_BAD_ARG); 55 CHK(ssp_ranst_discrete_setup(ran, NULL, nweights) == RES_BAD_ARG); 56 CHK(ssp_ranst_discrete_setup(NULL, weights, nweights) == RES_BAD_ARG); 57 CHK(ssp_ranst_discrete_setup(ran, weights, nweights) == RES_OK); 58 59 memset(tmp, 0, sizeof(tmp)); 60 FOR_EACH(i, 0, NSAMPS) { 61 const size_t k = ssp_ranst_discrete_get(rng, ran); 62 double pdf; 63 CHK(k < nweights); 64 pdf = ssp_ranst_discrete_pdf(k, ran); 65 ++tmp[k]; 66 CHK(pdf == weights[k]); 67 CHK(pdf >= 0.f); 68 CHK(pdf <= 1.f); 69 } 70 FOR_EACH(i, 0, nweights) { 71 CHK(tmp[i] != 0); 72 } 73 74 CHK(ssp_ranst_discrete_setup(ran, weights, nweights-1) == RES_OK); 75 FOR_EACH(i, 0, NSAMPS) { 76 const size_t k = ssp_ranst_discrete_get(rng, ran); 77 double pdf; 78 CHK(k < nweights-1); 79 pdf = ssp_ranst_discrete_pdf(k, ran); 80 CHK(pdf >= 0.f); 81 CHK(pdf <= 1.f); 82 } 83 accum = 0; 84 FOR_EACH(i, 0, nweights-1) accum += ssp_ranst_discrete_pdf(i, ran); 85 CHK(eq_eps(accum, 1, 1.e-8) == 1); 86 CHK(ssp_ranst_discrete_ref_put(ran) == RES_OK); 87 88 CHK(ssp_rng_ref_put(rng) == RES_OK); 89 check_memory_allocator(&allocator); 90 mem_shutdown_proxy_allocator(&allocator); 91 CHK(mem_allocated_size() == 0); 92 return 0; 93 } 94