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)