Пример #1
0
    def testControllerPosAlerts(self):
        plog = '%s/positions.csv' % os.path.dirname(__file__)
        elog = '%s/equity.csv' % os.path.dirname(__file__)
        rlog = '%s/returns.csv' % os.path.dirname(__file__)
        slog = '%s/summary.csv' % os.path.dirname(__file__)

        ctrl = ClenowController({ 'Ag-1' : ['RR'] }, plog, elog, rlog, summaryFile=slog)
        ctrl.run(datetime.datetime(2011,12,31),datetime.datetime(2012,8,1),datetime.datetime(2013,12,31))

        tlog = '%s/posalert.log' % os.path.dirname(__file__)
        ctrl.writePositionAlerts(tlog)

        self.assertTrue(test_util.file_compare('%s/posalert.reflog' % os.path.dirname(__file__), tlog))
        os.remove(tlog)

        tlog = '%s/trade7.log' % os.path.dirname(__file__)
        ctrl.writeAllTrades(tlog)
        
        self.assertTrue(test_util.file_compare('%s/trade7.reflog' % os.path.dirname(__file__), tlog))
        os.remove(tlog)
        
        tlog = '%s/tssb' % os.path.dirname(__file__)
        ctrl.writeTSSBTrades(tlog)

        self.assertTrue(test_util.file_compare('%s_long.reflog' % tlog, '%s_long.csv' % tlog))
        self.assertTrue(test_util.file_compare('%s_short.reflog' % tlog, '%s_short.csv' % tlog))
        os.remove('%s_long.csv' % tlog)
        os.remove('%s_short.csv' % tlog)
        os.remove(plog)
        os.remove(elog)
        os.remove(rlog)
        os.remove(slog)
Пример #2
0
    def testControllerPosAlerts(self):
        plog = '%s/positions.csv' % os.path.dirname(__file__)
        elog = '%s/equity.csv' % os.path.dirname(__file__)
        rlog = '%s/returns.csv' % os.path.dirname(__file__)
        slog = '%s/summary.csv' % os.path.dirname(__file__)

        ctrl = ClenowController({'Ag-1': ['RR']},
                                plog,
                                elog,
                                rlog,
                                summaryFile=slog)
        ctrl.run(datetime.datetime(2011, 12,
                                   31), datetime.datetime(2012, 8, 1),
                 datetime.datetime(2013, 12, 31))

        tlog = '%s/posalert.log' % os.path.dirname(__file__)
        ctrl.writePositionAlerts(tlog)

        self.assertTrue(
            test_util.file_compare(
                '%s/posalert.reflog' % os.path.dirname(__file__), tlog))
        os.remove(tlog)

        tlog = '%s/trade7.log' % os.path.dirname(__file__)
        ctrl.writeAllTrades(tlog)

        self.assertTrue(
            test_util.file_compare(
                '%s/trade7.reflog' % os.path.dirname(__file__), tlog))
        os.remove(tlog)

        tlog = '%s/tssb' % os.path.dirname(__file__)
        ctrl.writeTSSBTrades(tlog)

        self.assertTrue(
            test_util.file_compare('%s_long.reflog' % tlog,
                                   '%s_long.csv' % tlog))
        self.assertTrue(
            test_util.file_compare('%s_short.reflog' % tlog,
                                   '%s_short.csv' % tlog))
        os.remove('%s_long.csv' % tlog)
        os.remove('%s_short.csv' % tlog)
        os.remove(plog)
        os.remove(elog)
        os.remove(rlog)
        os.remove(slog)