def test_conf_file(self):
self.write_conf_file()
try:
args = parse_args(['-c', 'test.conf'])
self.assertEqual(args['algofile'], 'test.py')
self.assertEqual(args['symbols'], 'test_symbols')
self.assertEqual(args['start'], '1990-1-1')
self.assertEqual(args['end'], cli.DEFAULTS['end'])
finally:
os.remove('test.conf')
def test_conf_file(self):
self.write_conf_file()
try:
args = parse_args(["-c", "test.conf"])
self.assertEqual(args["algofile"], "test.py")
self.assertEqual(args["symbols"], "test_symbols")
self.assertEqual(args["start"], "1990-1-1")
self.assertEqual(args["data_frequency"], cli.DEFAULTS["data_frequency"])
finally:
os.remove("test.conf")
def test_overwrite(self):
self.write_conf_file()
try:
args = parse_args(["-c", "test.conf", "--start", "1992-1-1", "--algofile", "test2.py"])
# Overwritten values
self.assertEqual(args["algofile"], "test2.py")
self.assertEqual(args["start"], "1992-1-1")
# Non-overwritten values
self.assertEqual(args["symbols"], "test_symbols")
# Default values
self.assertEqual(args["data_frequency"], cli.DEFAULTS["data_frequency"])
finally:
os.remove("test.conf")
def test_overwrite(self):
self.write_conf_file()
try:
args = parse_args(['-c', 'test.conf', '--start', '1992-1-1',
'--algofile', 'test2.py'])
# Overwritten values
self.assertEqual(args['algofile'], 'test2.py')
self.assertEqual(args['start'], '1992-1-1')
# Non-overwritten values
self.assertEqual(args['symbols'], 'test_symbols')
# Default values
self.assertEqual(args['end'], cli.DEFAULTS['end'])
finally:
os.remove('test.conf')
#!/usr/bin/env python
#
# Copyright 2014 Quantopian, Inc.
#
# 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.
import logbook
import sys
from zipline.utils import parse_args, run_pipeline
if __name__ == "__main__":
logbook.StderrHandler().push_application()
parsed = parse_args(sys.argv[1:])
run_pipeline(print_algo=True, **parsed)
sys.exit(0)
def test_example_run_pipeline(self):
example = os.path.join(example_dir(), 'buyapple.py')
confs = ['-f', example, '--start', '2011-1-1', '--end', '2012-1-1']
parsed_args = parse_args(confs)
run_pipeline(**parsed_args)
def test_defaults(self):
args = parse_args([])
for k, v in iteritems(cli.DEFAULTS):
self.assertEqual(v, args[k])
def test_example_run_pipline(self):
example = os.path.join(example_dir(), "buyapple.py")
confs = ["-f", example, "--start", "2011-1-1", "--end", "2012-1-1"]
parsed_args = parse_args(confs)
run_pipeline(**parsed_args)
def test_example_run_pipeline(self):
example = os.path.join(example_dir(), 'buyapple.py')
confs = ['-f', example, '--start', '2011-1-1', '--end', '2012-1-1']
parsed_args = parse_args(confs)
run_pipeline(**parsed_args)
def test_defaults(self):
args = parse_args([])
for k, v in iteritems(cli.DEFAULTS):
self.assertEqual(v, args[k])
