runtests.py

# This file was originally part of PyAlgoTrade.
#
# Copyright 2011 Gabriel Martin Becedillas Ruiz
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
.. moduleauthor:: Gabriel Martin Becedillas Ruiz <gabriel.becedillas@gmail.com>
"""

import unittest
from testcases import technical_test
from testcases import dataseries_test
from testcases import csvbarfeed_test
from testcases import dbfeed_test
from testcases import broker_test
from testcases import strategy_test
#from testcases import smacrossover_strategy_test
#from testcases import multi_symbol_strategy_test
from testcases import talib_test
from testcases import observer_test
from testcases import returns_analyzer_test
from testcases import trades_analyzer_test
from testcases import sharpe_analyzer_test
from testcases import drawdown_analyzer_test
from testcases import utils_test
from testcases import doc_test

def getTestCases():
    ret = []
    ret += technical_test.getTestCases()
    ret += dataseries_test.getTestCases()
    ret += csvbarfeed_test.getTestCases()
    #ret += dbfeed_test.getTestCases()
    ret += broker_test.getTestCases()
    ret += strategy_test.getTestCases(includeExternal=False)
    #ret += smacrossover_strategy_test.getTestCases()
    #ret += multi_symbol_strategy_test.getTestCases()
    #ret += talib_test.getTestCases()
    ret += observer_test.getTestCases()
    ret += returns_analyzer_test.getTestCases()
    ret += trades_analyzer_test.getTestCases()
    ret += sharpe_analyzer_test.getTestCases()
    ret += drawdown_analyzer_test.getTestCases()
    ret += utils_test.getTestCases()
    ret += doc_test.getTestCases()

    return ret

def main():
    suite = unittest.TestSuite()
    suite.addTests(getTestCases())
    runner = unittest.TextTestRunner(verbosity=2)
    runner.run(suite)

def profile():
    import cProfile
    import pstats
    profFile = "prof"
    cProfile.run("main()", profFile)
    p = pstats.Stats(profFile)
    # p.dump_stats("runtests.profile")
    p.strip_dirs().sort_stats("time").print_stats()

if __name__ == "__main__":
    main()
    # profile()