#include <math.h>
#include <stddef.h>
#include "ts.h"

bool ts_test_atr_bands( void )
{
    fp_t const o[] = { 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 };
    fp_t const h[] = { 12, 13, 14, 15, 16, 17, 18, 19, 20, 21 };
    fp_t const l[] = { 9, 10, 11, 12, 13, 14, 15, 16, 17, 18 };
    fp_t const cp[] = { 11, 12, 13, 14, 15, 16, 17, 18, 19, 20 };

    fp_t const * const x[] = { o, h, l, cp };

    /* k=3, mult=2.0 */
    /* TR[1]=3, TR[2]=3, TR[3]=3 => ATR=3.0 */
    /* TR[4]=3, ... => ATR stays 3.0 */
    /* Middle(SMA3 of close from idx 1): (12+13+14)/3=13, then rolling */
    /* Middle: 13, 14, 15, 16, 17, 18, 19 */
    /* Upper: 19, 20, 21, 22, 23, 24, 25 */
    /* Lower: 7, 8, 9, 10, 11, 12, 13 */
    fp_t const em[] = { 13, 14, 15, 16, 17, 18, 19 };
    fp_t const eu[] = { 19, 20, 21, 22, 23, 24, 25 };
    fp_t const el[] = { 7, 8, 9, 10, 11, 12, 13 };
    fp_t const * const exp[3] = { eu, em, el };

    typedef struct { uint_t k; fp_t mult; } c_t;
    c_t c = { .k = 3, .mult = 2.0 };
    return ts_test(ts_atr_bands, ts_atr_bands_offset, x, exp, sizeof( o ) / sizeof( *o ), (ts_constant_t *)&c, &ts_atr_bands_p);
}