test_powerspectrum.c File Reference

#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <omp.h>
#include <math.h>
#include <mpi.h>
#include "stub.h"
#include <bigfile-mpi.h>
#include <libgadget/powerspectrum.h>

Include dependency graph for test_powerspectrum.c:

Go to the source code of this file.

Macros
#define	NUM_THREADS   4

Functions
static void	test_total_powerspectrum (void **state)
int	main (void)

Macro Definition Documentation

NUM_THREADS

#define NUM_THREADS   4

Definition at line 18 of file test_powerspectrum.c.

Function Documentation

main()

int main

(

void

)

Definition at line 60 of file test_powerspectrum.c.

{
    const struct CMUnitTest tests[] = {
        cmocka_unit_test(test_total_powerspectrum),
    };
    return cmocka_run_group_tests_mpi(tests, NULL, NULL);
}

References test_total_powerspectrum().

Here is the call graph for this function:

test_total_powerspectrum()

static void test_total_powerspectrum ( void **  state )

static

Definition at line 21 of file test_powerspectrum.c.

                                                   {
    (void) state;
    /*Check allocation*/
    int nmpi;
    Power PowerSpectrum;
    MPI_Comm_size(MPI_COMM_WORLD, &nmpi);
 
    powerspectrum_alloc(&PowerSpectrum,15,NUM_THREADS, 0, 3.085678e24);
    assert_true(PowerSpectrum.Nmodes);
    assert_true(PowerSpectrum.Power);
    assert_true(PowerSpectrum.kk);
    powerspectrum_zero(&PowerSpectrum);
    assert_true(PowerSpectrum.Nmodes[0] == 0);
    assert_true(PowerSpectrum.Nmodes[PowerSpectrum.size-1] == 0);
 
    //Construct input power (this would be done by the power spectrum routine in petapm)
    int ii, th;
    for(ii=0; ii<15; ii++) {
        for(th = 0; th < NUM_THREADS; th++) {
            PowerSpectrum.Nmodes[ii+PowerSpectrum.size*th] = ii;
            PowerSpectrum.Power[ii+PowerSpectrum.size*th] = ii*sin(ii)*sin(ii);
            PowerSpectrum.kk[ii+PowerSpectrum.size*th] = ii*ii;
        }
    }
    PowerSpectrum.Norm = 1;
    /*Now every thread and every MPI has the same data. Sum it.*/
    powerspectrum_sum(&PowerSpectrum);
 
    /*Check summation was done correctly*/
    assert_true(PowerSpectrum.Nmodes[0] == NUM_THREADS*nmpi);
    assert_true(PowerSpectrum.Nmodes[13] == NUM_THREADS*nmpi*14);
 
    assert_true(fabs(PowerSpectrum.Power[0] - sin(1)*sin(1)) < 1e-5);
    assert_true(fabs(PowerSpectrum.Power[12] - sin(13)*sin(13)) < 1e-5);
    assert_true(fabs(PowerSpectrum.kk[12] - 2 * M_PI *13) < 1e-5);
    assert_true(fabs(PowerSpectrum.kk[0] - 2 * M_PI ) < 1e-5);
 
}

References _powerspectrum::kk, _powerspectrum::Nmodes, _powerspectrum::Norm, NUM_THREADS, _powerspectrum::Power, powerspectrum_alloc(), powerspectrum_sum(), powerspectrum_zero(), and _powerspectrum::size.

Referenced by main().

Here is the call graph for this function:

Here is the caller graph for this function: