def testCompInv_1(self): files = [os.path.join("samples", "data", src) for src in ["aeti-2011-yahoofinance.csv", "egan-2011-yahoofinance.csv", "simo-2011-yahoofinance.csv", "glng-2011-yahoofinance.csv"]] with common.CopyFiles(files, "."): lines = common.run_sample_script("compinv-1.py") # Skip the first two lines that have debug messages from the # broker. self.assertTrue(common.compare_head("compinv-1.output", lines[2:]))
def testTutorial4(self): with common.CopyFiles( [os.path.join("testcases", "data", "orcl-2000.csv")], "."): res = common.run_sample_script("tutorial-4.py") self.assertTrue( common.compare_head("tutorial-4.output", res.get_output_lines(True))) self.assertTrue(res.exit_ok())
def testSampleStrategyAnalyzer(self): with common.CopyFiles( [os.path.join("testcases", "data", "orcl-2000.csv")], "."): lines = common.run_sample_script( "sample-strategy-analyzer.py").split("\n") self.assertTrue( common.compare_head("sample-strategy-analyzer.output", lines[:-1]))
def testTutorial3(self): with common.CopyFiles( [os.path.join("testcases", "data", "orcl-2000.csv")], "."): lines = common.run_sample_script("tutorial-3.py").split("\n") self.assertTrue( common.compare_head("tutorial-3.output", lines[:30])) self.assertTrue( common.compare_tail("tutorial-3.output", lines[-4:-1]))
def testSampleStrategyAnalyzer(self): with common.CopyFiles( [os.path.join("testcases", "data", "orcl-2000.csv")], "."): res = common.run_sample_script("sample-strategy-analyzer.py") self.assertTrue( common.compare_head("sample-strategy-analyzer.output", res.get_output_lines(True))) self.assertTrue(res.exit_ok())
def testCSVFeed(self): with common.CopyFiles([os.path.join("samples", "data", "quandl_gold_2.csv")], "."): code = """import sys sys.path.append('samples') import csvfeed_1 """ lines = common.run_python_code(code).split("\n") self.assertTrue(common.compare_head("csvfeed_1.output", lines[0:10])) self.assertTrue(common.compare_tail("csvfeed_1.output", lines[-10:-1]))
def testExample2(self): with common.CopyFiles([os.path.join("testcases", "data", "30min-bitstampUSD-2.csv")], "30min-bitstampUSD.csv"): code = """import sys sys.path.append('samples') import bccharts_example_2 bccharts_example_2.main(False) """ lines = common.run_python_code(code).split("\n") self.assertTrue(common.compare_head("bccharts_example_2.output", lines[0:10], "testcases/data")) self.assertTrue(common.compare_tail("bccharts_example_2.output", lines[-10:-1], "testcases/data"))
def testTutorial1MtGox(self): with common.CopyFiles([os.path.join("samples", "data", "trades-mtgox-usd-2013-03.csv")], "."): code = """import sys sys.path.append('samples') import tutorial_mtgox_1 tutorial_mtgox_1.main(False) """ lines = common.run_python_code(code).split("\n") self.assertTrue(common.compare_head("tutorial_mtgox_1.output", lines[0:10])) self.assertTrue(common.compare_tail("tutorial_mtgox_1.output", lines[-10:-1]))
def testCompInv_1(self): files = [ os.path.join("samples", "data", src) for src in [ "aeti-2011-yahoofinance.csv", "egan-2011-yahoofinance.csv", "simo-2011-yahoofinance.csv", "glng-2011-yahoofinance.csv" ] ] with common.CopyFiles(files, "."): lines = common.run_sample_script("compinv-1.py").split("\n") self.assertTrue(common.compare_head("compinv-1.output", lines[:-1]))
def testExample1(self): with common.CopyFiles([os.path.join("testcases", "data", "bitstampUSD-2.csv")], "bitstampUSD.csv"): code = """import sys sys.path.append('samples') import bccharts_example_1 bccharts_example_1.main() """ common.run_python_code(code) lines = common.get_file_lines("30min-bitstampUSD.csv") self.assertTrue(common.compare_head("30min-bitstampUSD-2.csv", lines[0:10], "testcases/data")) self.assertTrue(common.compare_tail("30min-bitstampUSD-2.csv", lines[-10:], "testcases/data")) os.remove("30min-bitstampUSD.csv")
def testTutorial1MtGox(self): with common.CopyFiles( [os.path.join("samples", "data", "trades-mtgox-usd-2013-03.csv")], "."): code = """import sys sys.path.append('samples') import tutorial_mtgox_1 tutorial_mtgox_1.main(False) """ lines = common.run_python_code(code).split("\n") self.assertTrue( common.compare_head("tutorial_mtgox_1.output", lines[0:10])) self.assertTrue( common.compare_tail("tutorial_mtgox_1.output", lines[-10:-1]))
def testQuandl(self): files = [] for year in range(2006, 2013): for symbol in ["GORO"]: fileName = "WIKI-%s-%d-quandl.csv" % (symbol, year) files.append(os.path.join("samples", "data", fileName)) files.append(os.path.join("samples", "data", "quandl_gold_2.csv")) with common.CopyFiles(files, "."): code = """import sys sys.path.append('samples') import quandl_sample quandl_sample.main(False) """ lines = common.run_python_code(code).split("\n") self.assertTrue(common.compare_head("quandl_sample.output", lines[0:10])) self.assertTrue(common.compare_tail("quandl_sample.output", lines[-10:-1]))
def testTechnical_1(self): lines = common.run_sample_script("technical-1.py") self.assertTrue(common.compare_head("technical-1.output", lines))
def testCompInv_1(self): files = [os.path.join("samples", "data", src) for src in ["aeti-2011-yahoofinance.csv", "egan-2011-yahoofinance.csv", "simo-2011-yahoofinance.csv", "glng-2011-yahoofinance.csv"]] with common.CopyFiles(files, "."): lines = common.run_sample_script("compinv-1.py").split("\n") self.assertTrue(common.compare_head("compinv-1.output", lines[:-1]))
def testTechnical_1(self): lines = common.run_sample_script("technical-1.py").split("\n") self.assertTrue(common.compare_head("technical-1.output", lines[:-1]))
def testSampleStrategyAnalyzer(self): with common.CopyFiles([os.path.join("testcases", "data", "orcl-2000.csv")], "."): lines = common.run_sample_script("sample-strategy-analyzer.py").split("\n") self.assertTrue(common.compare_head("sample-strategy-analyzer.output", lines[:-1]))
def testTutorial3(self): with common.CopyFiles([os.path.join("testcases", "data", "orcl-2000.csv")], "."): lines = common.run_sample_script("tutorial-3.py") self.assertTrue(common.compare_head("tutorial-3.output", lines[:30])) self.assertTrue(common.compare_tail("tutorial-3.output", lines[-3:]))
def testSampleStrategyAnalyzer(self): with common.CopyFiles([os.path.join("testcases", "data", "orcl-2000.csv")], "."): res = common.run_sample_script("sample-strategy-analyzer.py") self.assertTrue(common.compare_head("sample-strategy-analyzer.output", res.get_output_lines(True))) self.assertTrue(res.exit_ok())
def testTutorial4(self): with common.CopyFiles([os.path.join("testcases", "data", "orcl-2000.csv")], "."): res = common.run_sample_script("tutorial-4.py") self.assertTrue(common.compare_head("tutorial-4.output", res.get_output_lines(True))) self.assertTrue(res.exit_ok())
def testTechnical_1(self): res = common.run_sample_script("technical-1.py") self.assertTrue( common.compare_head("technical-1.output", res.get_output_lines(True))) self.assertTrue(res.exit_ok())
def testTechnical_1(self): res = common.run_sample_script("technical-1.py") self.assertTrue(common.compare_head("technical-1.output", res.get_output_lines(True))) self.assertTrue(res.exit_ok())
def testTutorial4(self): with common.CopyFiles([os.path.join("testcases", "data", "orcl-2000.csv")], "."): lines = common.run_sample_script("tutorial-4.py").split("\n") self.assertTrue(common.compare_head("tutorial-4.output", lines[:-1]))