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