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 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 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 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]))
