def test_env_buy(self):
trade = Trade()
trade.new_episode()
print(trade.time)
trade.step(ACTION.BUY)
print(trade.time)
def test_read_one_episode(self):
trade = Trade()
trade.new_episode()
num_of_records = 0
for i in range(1000):
if trade.new_sec():
self.assertIsNotNone(trade.boards)
self.assertIsNotNone(trade.buy_trade_price)
self.assertIsNotNone(trade.sell_trade_price)
self.assertIsNotNone(trade.buy_book_price)
self.assertIsNotNone(trade.sell_book_vol)
self.assertIsNotNone(trade.sell_book_price)
self.assertIsNotNone(trade.buy_book_vol)
self.assertIsNotNone(trade.market_sell_price)
self.assertIsNotNone(trade.buy_trade_vol)
self.assertIsNotNone(trade.fix_buy_price)
self.assertIsNotNone(trade.market_buy_price)
self.assertIsNotNone(trade.time)
num_of_records += 1
print(trade.time)
else:
break
print(num_of_records)
def test_read_one_episode(self):
trade = Trade()
trade.new_episode()
num_of_records = 0
while trade.new_sec():
self.assertIsNotNone(trade.board)
self.assertIsNotNone(trade.buy_trade_price)
self.assertIsNotNone(trade.sell_trade_price)
self.assertIsNotNone(trade.buy_book_price)
self.assertIsNotNone(trade.sell_book_vol)
self.assertIsNotNone(trade.sell_book_price)
self.assertIsNotNone(trade.buy_book_vol)
self.assertIsNotNone(trade.market_sell_price)
self.assertIsNotNone(trade.buy_trade_vol)
self.assertIsNotNone(trade.fix_buy_price)
self.assertIsNotNone(trade.market_buy_price)
self.assertIsNotNone(trade.time)
print(trade.time)
num_of_records += 1
print(num_of_records)
def test_env_step(self):
trade = Trade()
trade.new_episode()
trade.step(ACTION.NOP)
trade.step(ACTION.BUY)
trade.step(ACTION.BUY_NOW)
trade.step(ACTION.SELL)
trade.step(ACTION.SELL_NOW)
def test_buy_order(self):
trade = Trade()
trade.new_episode()
print(trade.time)
trade.step(ACTION.BUY)
print(trade.time)
print(trade.buy_order_price)
print(trade.margin)
def test_sell_order(self):
trade = Trade()
trade.new_episode()
print(trade.time)
trade.step(ACTION.SELL)
print(trade.time)
print(trade.sell_order_price)
print(trade.margin)
def test_new_episode(self):
trade = Trade()
trade.new_episode()
result = trade.new_sec()
print(result)
print(trade.time)
def test_env_sell_buy(self):
trade = Trade()
trade.new_episode()
print(trade.time)
trade.step(ACTION.SELL_NOW)
print(trade.time)
print(trade.sell_order_price)
print(trade.margin)
# close transaction
print(trade.time)
trade.step(ACTION.BUY_NOW)
print(trade.time)
print(trade.margin)
def test_new_sec(self):
trade = Trade()
trade.new_episode()
trade.new_sec()
print(trade.time)
trade.new_sec()
print(trade.time)
trade.new_sec()
print(trade.time)
trade.new_sec()
print(trade.time)
def one_episode(self, env: Trade):
s = env.new_episode()
while True:
action = self.policy(s)
print('action->', action, end='')
next_state, reward, done, info = env.step(action)
print('reward->', reward)
if done:
break
s = next_state
return reward
def test_actions(self):
trade = Trade()
trade.new_episode()
print(trade.action_space)
