Python head_file Examples

Example #1

File: doc_test.py Project: berlm/pyalgotrade

 def testTechnical_1(self):
     res = common.run_sample_script("technical-1.py")
     lines = res.get_output_lines()
     self.assertEqual(
         lines,
         common.head_file("technical-1.output", len(lines), path="../samples")
     )

Example #2

    def testTechnical_1(self):
        res = common.run_sample_script("technical-1.py")

        self.assertTrue(res.exit_ok())
        lines = res.get_output_lines()
        self.assertEqual(lines,
                         common.head_file("technical-1.output", len(lines)))

Example #3

    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()
"""
            res = common.run_python_code(code)
            lines = common.get_file_lines("30min-bitstampUSD.csv")
            os.remove("30min-bitstampUSD.csv")

            # Float str doesn't behave the same in Pyhton2 and Python3.
            if six.PY3:
                lines = lines[0:1] + [
                    round_csv_line(line, 5, 8) for line in lines[1:]
                ]

            self.assertTrue(res.exit_ok(), res.get_output())
            self.assertEqual(
                lines[0:10],
                common.head_file("30min-bitstampUSD-2.csv",
                                 10,
                                 path="testcases/data"))
            self.assertEqual(
                lines[-10:],
                common.tail_file("30min-bitstampUSD-2.csv",
                                 10,
                                 path="testcases/data"))

Example #4

 def testTutorial4(self):
     with common.CopyFiles(
         [os.path.join("testcases", "data", "orcl-2000.csv")], "."):
         res = common.run_sample_script("tutorial-4.py")
         lines = res.get_output_lines(True)
         self.assertEqual(common.head_file("tutorial-4.output", len(lines)),
                          lines)
         self.assertTrue(res.exit_ok())

Example #5

 def testTutorial3(self):
     with common.CopyFiles(
         [os.path.join("testcases", "data", "orcl-2000.csv")], "."):
         res = common.run_sample_script("tutorial-3.py")
         self.assertEqual(common.head_file("tutorial-3.output", 30),
                          res.get_output_lines(True)[:30])
         self.assertEqual(common.tail_file("tutorial-3.output", 3),
                          res.get_output_lines(True)[-3:])
         self.assertTrue(res.exit_ok())

Example #6

File: doc_test.py Project: gansaihua/pyalgotrade

    def testTechnical_1(self):
        res = common.run_sample_module("technical-1")

        self.assertTrue(res.exit_ok(), res.get_output())
        lines = res.get_output_lines()
        self.assertEqual(
            lines,
            common.head_file("technical-1.output", len(lines))
        )

Example #7

File: doc_test.py Project: gansaihua/pyalgotrade

 def testTutorial4(self):
     with common.CopyFiles([os.path.join("testcases", "data", "WIKI-ORCL-2000-quandl.csv")], "."):
         res = common.run_sample_module("tutorial-4")
         lines = res.get_output_lines(True)
         self.assertEqual(
             common.head_file("tutorial-4.output", len(lines)),
             lines
         )
         self.assertTrue(res.exit_ok())

Example #8

 def testTutorial4(self):
     with common.CopyFiles(
         [os.path.join("testcases", "data", "WIKI-ORCL-2000-quandl.csv")],
             "."):
         res = common.run_sample_module("tutorial-4")
         lines = res.get_output_lines(True)
         self.assertEqual(common.head_file("tutorial-4.output", len(lines)),
                          lines, res.get_output())
         self.assertTrue(res.exit_ok())

Example #9

File: doc_test.py Project: berlm/pyalgotrade

    def testSampleStrategyAnalyzer(self):
        with common.CopyFiles([os.path.join("data", "orcl-2000.csv")], "."):
            res = common.run_sample_script("sample-strategy-analyzer.py")

            lines = res.get_output_lines()
            self.assertEqual(
                lines,
                common.head_file("sample-strategy-analyzer.output", len(lines), path="../samples")
            )

Example #10

File: doc_test.py Project: Chin-I/pyalgotrade

    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(res.exit_ok())
            lines = res.get_output_lines()
            self.assertEqual(
                lines,
                common.head_file("sample-strategy-analyzer.output", len(lines))
            )

Example #11

 def testTutorial3(self):
     with common.CopyFiles(
         [os.path.join("testcases", "data", "WIKI-ORCL-2000-quandl.csv")],
             "."):
         res = common.run_sample_module("tutorial-3")
         self.assertEqual(common.head_file("tutorial-3.output", 30),
                          res.get_output_lines(True)[:30], res.get_output())
         self.assertEqual(common.tail_file("tutorial-3.output", 3),
                          res.get_output_lines(True)[-3:], res.get_output())
         self.assertTrue(res.exit_ok())

Example #12

    def testSampleStrategyAnalyzer(self):
        with common.CopyFiles([os.path.join("testcases", "data", "orcl-2000-yahoofinance.csv")], "."):
            res = common.run_sample_module("sample-strategy-analyzer")

            self.assertTrue(res.exit_ok())
            lines = res.get_output_lines()
            self.assertGreaterEqual(len(lines), 20)
            self.assertEqual(
                lines,
                common.head_file("sample-strategy-analyzer.output", len(lines))
            )

Example #13

File: doc_test.py Project: berlm/pyalgotrade

 def testTutorial2(self):
     with common.CopyFiles([os.path.join("data", "orcl-2000.csv")], "."):
         res = common.run_sample_script("tutorial-2.py")
         self.assertEqual(
             common.head_file("tutorial-2.output", 15, path="../samples"),
             res.get_output_lines(True)[:15]
         )
         self.assertEqual(
             common.tail_file("tutorial-2.output", 3, path="../samples"),
             res.get_output_lines(True)[-3:]
         )

Example #14

File: doc_test.py Project: berlm/pyalgotrade

 def testCSVFeed(self):
     with common.CopyFiles([os.path.join("..", "samples", "data", "quandl_gold_2.csv")], "."):
         res = common.run_sample_script("csvfeed_1.py")
         lines = res.get_output_lines(True)
         self.assertEqual(
             common.head_file("csvfeed_1.output", 10, path="../samples"),
             lines[0:10]
         )
         self.assertEqual(
             common.tail_file("csvfeed_1.output", 10, path="../samples"),
             lines[-10:]
         )

Example #15

File: doc_test.py Project: berlm/pyalgotrade

 def testExample2(self):
     with common.CopyFiles([os.path.join("data", "30min-bitstampUSD-2.csv")], "30min-bitstampUSD.csv"):
         from samples import bccharts_example_2
         res = bccharts_example_2.main(False)
         self.assertEqual(
             res.get_output_lines()[0:10],
             common.head_file("bccharts_example_2.output", 10, path="data")
         )
         self.assertEqual(
             res.get_output_lines()[-10:],
             common.tail_file("bccharts_example_2.output", 10, path="data")
         )

Example #16

    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, "."):
            res = common.run_sample_script("compinv-1.py")

            self.assertTrue(res.exit_ok())
            # Skip the first two lines that have debug messages from the broker.
            lines = res.get_output_lines()[2:]
            self.assertEqual(
                lines,
                common.head_file("compinv-1.output", len(lines))
            )

Example #17

File: doc_test.py Project: Chin-I/pyalgotrade

 def testTutorial2(self):
     with common.CopyFiles([os.path.join("testcases", "data", "orcl-2000.csv")], "."):
         res = common.run_sample_script("tutorial-2.py")
         self.assertEqual(
             common.head_file("tutorial-2.output", 15),
             res.get_output_lines(True)[:15]
         )
         self.assertEqual(
             common.tail_file("tutorial-2.output", 3),
             res.get_output_lines(True)[-3:]
         )
         self.assertTrue(res.exit_ok())

Example #18

File: doc_test.py Project: gansaihua/pyalgotrade

 def testTutorial3(self):
     with common.CopyFiles([os.path.join("testcases", "data", "WIKI-ORCL-2000-quandl.csv")], "."):
         res = common.run_sample_module("tutorial-3")
         self.assertEqual(
             common.head_file("tutorial-3.output", 30),
             res.get_output_lines(True)[:30]
         )
         self.assertEqual(
             common.tail_file("tutorial-3.output", 3),
             res.get_output_lines(True)[-3:]
         )
         self.assertTrue(res.exit_ok())

Example #19

    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
"""
            res = common.run_python_code(code)
            lines = res.get_output_lines(True)
            self.assertEqual(common.head_file("csvfeed_1.output", 10),
                             lines[0:10])
            self.assertEqual(common.tail_file("csvfeed_1.output", 10),
                             lines[-10:])
            self.assertTrue(res.exit_ok())

Example #20

    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)
"""
            res = common.run_python_code(code)
            self.assertTrue(res.exit_ok(), res.get_output())
            self.assertEqual(res.get_output_lines()[0:10],
                             common.head_file("bccharts_example_2.output", 10))
            self.assertEqual(res.get_output_lines()[-10:],
                             common.tail_file("bccharts_example_2.output", 10))

Example #21

File: doc_test.py Project: berlm/pyalgotrade

    def testExample1(self):
        with common.CopyFiles([os.path.join("data", "bitstampUSD-2.csv")], "bitstampUSD.csv"):
            from samples import bccharts_example_1
            bccharts_example_1.main()

            lines = common.get_file_lines("30min-bitstampUSD.csv")
            os.remove("30min-bitstampUSD.csv")

            self.assertEqual(
                lines[0:10],
                common.head_file("30min-bitstampUSD-2.csv", 10, path="data")
            )
            self.assertEqual(
                lines[-10:],
                common.tail_file("30min-bitstampUSD-2.csv", 10, path="data")
            )

Example #22

File: doc_test.py Project: gansaihua/pyalgotrade

    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
"""
            res = common.run_python_code(code)
            lines = res.get_output_lines(True)
            self.assertEqual(
                common.head_file("csvfeed_1.output", 10),
                lines[0:10]
            )
            self.assertEqual(
                common.tail_file("csvfeed_1.output", 10),
                lines[-10:]
            )
            self.assertTrue(res.exit_ok())

Example #23

File: doc_test.py Project: gansaihua/pyalgotrade

    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)
"""
            res = common.run_python_code(code)
            self.assertTrue(res.exit_ok(), res.get_output())
            self.assertEqual(
                res.get_output_lines()[0:10],
                common.head_file("bccharts_example_2.output", 10, path="testcases/data")
            )
            self.assertEqual(
                res.get_output_lines()[-10:],
                common.tail_file("bccharts_example_2.output", 10, path="testcases/data")
            )

Example #24

    def testCompInv_1(self):
        files = [os.path.join("samples", "data", src) for src in [
            "WIKI-IBM-2011-quandl.csv",
            "WIKI-AES-2011-quandl.csv",
            "WIKI-AIG-2011-quandl.csv",
            "WIKI-ORCL-2011-quandl.csv",
        ]]

        with common.CopyFiles(files, "."):
            res = common.run_sample_module("compinv-1")

            self.assertTrue(res.exit_ok())
            # Skip the first two lines that have debug messages from the broker.
            lines = res.get_output_lines()
            self.assertEqual(
                lines,
                common.head_file("compinv-1.output", len(lines))
            )

Example #25

File: doc_test.py Project: berlm/pyalgotrade

    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, "."):
            from samples import quandl_sample
            res = quandl_sample.main(False)
            self.assertEqual(
                res.get_output_lines()[0:10],
                common.head_file("quandl_sample.output", 10, path="../samples")
            )
            self.assertEqual(
                res.get_output_lines()[-10:],
                common.tail_file("quandl_sample.output", 10, path="../samples")
            )

Example #26

    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)
"""
            res = common.run_python_code(code)
            self.assertTrue(res.exit_ok())
            self.assertEqual(res.get_output_lines()[0:10],
                             common.head_file("quandl_sample.output", 10))
            self.assertEqual(res.get_output_lines()[-10:],
                             common.tail_file("quandl_sample.output", 10))

Example #27

    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()
"""
            res = common.run_python_code(code)
            lines = common.get_file_lines("30min-bitstampUSD.csv")
            os.remove("30min-bitstampUSD.csv")

            self.assertTrue(res.exit_ok())
            self.assertEqual(
                lines[0:10],
                common.head_file("30min-bitstampUSD-2.csv", 10, path="testcases/data")
            )
            self.assertEqual(
                lines[-10:],
                common.tail_file("30min-bitstampUSD-2.csv", 10, path="testcases/data")
            )

Example #28

File: doc_test.py Project: gansaihua/pyalgotrade

    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()
"""
            res = common.run_python_code(code)
            lines = common.get_file_lines("30min-bitstampUSD.csv")
            os.remove("30min-bitstampUSD.csv")

            # Float str doesn't behave the same in Pyhton2 and Python3.
            if six.PY3:
                lines = lines[0:1] + [round_csv_line(line, 5, 8) for line in lines[1:]]

            self.assertTrue(res.exit_ok())
            self.assertEqual(
                lines[0:10],
                common.head_file("30min-bitstampUSD-2.csv", 10, path="testcases/data")
            )
            self.assertEqual(
                lines[-10:],
                common.tail_file("30min-bitstampUSD-2.csv", 10, path="testcases/data")
            )

Example #29

File: doc_test.py Project: gansaihua/pyalgotrade

    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)
"""
            res = common.run_python_code(code)
            self.assertTrue(res.exit_ok())
            self.assertEqual(
                res.get_output_lines()[0:10],
                common.head_file("quandl_sample.output", 10)
            )
            self.assertEqual(
                res.get_output_lines()[-10:],
                common.tail_file("quandl_sample.output", 10)
            )