test_SE_Fixed.cpp

浏览该文件的文档.

/*
 * test_Signal.cpp
 *
 *  Created on: 2013-3-11
 *      Author: fasiondog
 */


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

#include <hikyuu/StockManager.h>
#include <hikyuu/trade_sys/system/crt/SYS_Simple.h>
#include <hikyuu/trade_sys/selector/crt/SE_Fixed.h>

using namespace hku;

BOOST_AUTO_TEST_CASE( test_SE_Fixed) {
    StockManager& sm = StockManager::instance();

    SYSPtr sys = SYS_Simple();
    SEPtr se = SE_Fixed();

    se->addStock(Stock(), sys);
    SystemList result = se->getSelectedSystemList(Datetime(200001010000L));
    BOOST_CHECK(result.size() == 0);

    se->addStock(sm["sh600000"], SYSPtr());
    result = se->getSelectedSystemList(Datetime(200001010000L));
    BOOST_CHECK(result.size() == 0);

    se->addStock(sm["sh600000"], sys);
    se->addStock(sm["sz000001"], sys);
    se->addStock(sm["sz000002"], sys);

    se->reset();
    result = se->getSelectedSystemList(Datetime(200001010000L));
    BOOST_CHECK(result.size() == 3);
    BOOST_CHECK(sm["sh600000"] == result[0]->getStock());
    BOOST_CHECK(sm["sz000001"] == result[1]->getStock());
    BOOST_CHECK(sm["sz000002"] == result[2]->getStock());

    se->clear();
    result = se->getSelectedSystemList(Datetime(200001010000L));
    BOOST_CHECK(result.size() == 0);

    se->addStock(sm["sh600000"], sys);
    se->addStock(sm["sz000001"], sys);
    se->addStock(sm["sz000002"], sys);
    se->reset();
    result = se->getSelectedSystemList(Datetime(200001010000L));
    BOOST_CHECK(result.size() == 3);
    BOOST_CHECK(sm["sh600000"] == result[0]->getStock());
    BOOST_CHECK(sm["sz000001"] == result[1]->getStock());
    BOOST_CHECK(sm["sz000002"] == result[2]->getStock());

    SEPtr se2;
    se2 = se->clone();
    BOOST_CHECK(se2.get() != se.get());
    result = se2->getSelectedSystemList(Datetime(200001010000L));
    BOOST_CHECK(result.size() == 3);
    BOOST_CHECK(sm["sh600000"] == result[0]->getStock());
    BOOST_CHECK(sm["sz000001"] == result[1]->getStock());
    BOOST_CHECK(sm["sz000002"] == result[2]->getStock());
}