test_PRICELIST.cpp

浏览该文件的文档.

/*
 * test_PRICELIST.cpp
 *
 *  Created on: 2013-2-14
 *      Author: fasiondog
 */

#ifdef TEST_ALL_IN_ONE
    #include <boost/test/unit_test.hpp>
#else
    #define BOOST_TEST_MODULE test_hikyuu_indicator_suite
    #include <boost/test/unit_test.hpp>
#endif

#include <hikyuu/StockManager.h>
#include <hikyuu/indicator/crt/PRICELIST.h>
#include <hikyuu/indicator/crt/KDATA.h>
#include <hikyuu/indicator/crt/MA.h>

using namespace hku;

BOOST_AUTO_TEST_CASE( test_PRICELIST ) {
    PriceList tmp_list;
    Indicator result;

    result = PRICELIST(tmp_list);
    BOOST_CHECK(result.size() == tmp_list.size());
    BOOST_CHECK(result.empty() == true);

    for (size_t i = 0; i < 10; ++i) {
        tmp_list.push_back(i);
    }
    result = PRICELIST(tmp_list);
    BOOST_CHECK(result.size() == tmp_list.size());
    BOOST_CHECK(result.empty() == false);
    for (size_t i = 0; i < 10; ++i) {
        BOOST_CHECK(result[i] == tmp_list[i]);
    }

    price_t *p_tmp = NULL;
    result = PRICELIST(p_tmp, 10);
    BOOST_CHECK(result.size() == 0);
    BOOST_CHECK(result.empty() == true);

    price_t tmp[10];
    for (size_t i = 0; i < 10; ++i) {
        tmp[i] = i;
    }
    result = PRICELIST(tmp, 10);
    for (size_t i = 0; i < 10; ++i) {
        BOOST_CHECK(result[i] == tmp[i]);
    }

    result = PRICELIST(tmp_list);
    BOOST_CHECK(result.size() == 10);
    for (size_t i = 0; i < 10; ++i) {
        BOOST_CHECK(result[i] == tmp_list[i]);
    }

    result = PRICELIST(tmp_list, 1);
    BOOST_CHECK(result.size() == 10);
    BOOST_CHECK(result.discard() == 1);
    BOOST_CHECK(result[0] == Null<price_t>());
    for (size_t i = 1; i < 10; ++i) {
        BOOST_CHECK(result[i] == tmp_list[i]);
    }

    result = PRICELIST(Indicator());
    BOOST_CHECK(result.size() == 0);

    StockManager& sm = StockManager::instance();
    Stock stock = sm.getStock("sh000001");
    KQuery query(0, 30);
    KData kdata = stock.getKData(query);
    Indicator ikdata = KDATA(kdata);
    result = PRICELIST(ikdata);
    BOOST_CHECK(result.size() == ikdata.size());
    BOOST_CHECK(result.discard() == ikdata.discard());
    for (size_t i = 0; i < result.size(); ++i) {
        BOOST_CHECK(result[i] == ikdata[i]);
    }

    result = PRICELIST(ikdata, 1);
    BOOST_CHECK(result.size() == ikdata.size());
    BOOST_CHECK(result.discard() == ikdata.discard());
    for (size_t i = 0; i < result.size(); ++i) {
        BOOST_CHECK(result[i] == ikdata.get(i, 1));
    }
}