def test_computeSimpleMeanAverage_emptyListCase():
transaction = Transaction()
values = []
assert transaction.computeSimpleMeanAverage(values) == 0
def test_computeSimpleMeanAverage_normalCase():
transaction = Transaction()
values = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
assert transaction.computeSimpleMeanAverage(values) == 5
def test_sell_normalCase(capsys):
transaction = Transaction()
transaction.sell("USDT", "ETH", 2)
redir = capsys.readouterr()
assert redir.out == "sell USDT_ETH 2"
def test_buy_normalCase(capsys):
transaction = Transaction()
transaction.buy("USDT", "ETH", 2)
redir = capsys.readouterr()
assert redir.out == "buy USDT_ETH 2"
def test_no_moves_normalCase(capsys):
transaction = Transaction()
transaction.no_moves()
redir = capsys.readouterr()
assert redir.out == "pass\n"
def test_comput_standarDeviation_normalCase():
transaction = Transaction()
values = [5, 1, 2, 1, 5, 1, 2, 1, 5, 1, 2]
ret = transaction.comput_standarDeviation(values)
assert ret == 1.6663911618021237
def __init__(self):
self._settings = {}
self._candles = []
self._stack = Stack()
# Attributes to other classes
self._utils = Utilities()
self._t = Transaction()
# Run program's main loop
self.run()
def test_update_lastClosePrices_normalCase():
transaction = Transaction()
# Change `._period` to simplify tests.
transaction._period = 3
candles = [
Candle(
"BTC_ETH,1516219200,0.09075,0.08738895,0.0883,0.08921392,191.335761;USDT_ETH,1516219200,975,909.1676282,924.00000003,962.49435604,1282237.7927758;USDT_BTC,1516219200,10840.99999999,10350,10468.953,10768.92744569,4549482.7666951"
),
Candle(
"BTC_ETH,1516221000,0.091,0.0887794,0.08932623,0.09027925,103.06952136;USDT_ETH,1516221000,996.65,944.49435604,962.49435604,995,1169925.471947;USDT_BTC,1516221000,11800,10540.00922298,10701.00000006,11023.90000003,5869606.1180224"
),
Candle(
"BTC_ETH,1516222800,0.09229306,0.0895,0.09039,0.09075563,219.18672545;USDT_ETH,1516222800,996.65,964.60679991,995,990.00000001,1762788.8665345;USDT_BTC,1516222800,11173.94628473,10565.00000001,11028.47418901,10899.99999986,3176851.6894339"
),
Candle(
"BTC_ETH,1516224600,0.0927,0.09012191,0.0914842,0.09239994,321.10136768;USDT_ETH,1516224600,1060.99877995,985,990.5,1055.00000003,1946421.3026575;USDT_BTC,1516224600,11750,10845.55000011,10899.99999987,11725.98,3953421.7861957"
),
Candle(
"BTC_ETH,1516226400,0.094,0.09,0.09239995,0.09119839,475.95416552;USDT_ETH,1516226400,1075,996,1055.00000003,1019.4,2738163.4159786;USDT_BTC,1516226400,11793.08,10999.99999999,11730.99,11302,9127973.011532"
),
Candle(
"BTC_ETH,1516228200,0.0927,0.09,0.09142738,0.0907592,187.20202469;USDT_ETH,1516228200,1050,963.055579,1019.4,979.56000012,1668904.4017084;USDT_BTC,1516228200,11500.00000001,10811.166526,11305,10959.36154234,9590619.9055415"
),
Candle(
"BTC_ETH,1516230000,0.09077734,0.089,0.0902025,0.09042024,391.03913012;USDT_ETH,1516230000,1014.53857185,947.99999997,988.65893444,990.00000004,2494607.092476;USDT_BTC,1516230000,11176.31348786,10642.09801453,10959.36154234,10982.89484619,5262360.6032129"
)
]
expected = [[0.09119839, 0.0907592, 0.09042024],
[1019.4, 979.56000012, 990.00000004],
[11302, 10959.36154234, 10982.89484619]]
for candle in candles:
transaction.update_lastClosePrices(candle)
assert len(expected) == len(transaction._period_lastClosePrices)
for i in range(len(expected)):
assert transaction._period_lastClosePrices[i] == expected[i]
def test_isValidCurrency_wrongCase():
transaction = Transaction()
assert not transaction.isValidCurrency("ETHH")
def test_isValidCurrency_normalCase():
transaction = Transaction()
assert transaction.isValidCurrency("ETH")
class Trade():
"""
Main class of the Trade project.
"""
def __init__(self):
self._settings = {}
self._candles = []
self._stack = Stack()
# Attributes to other classes
self._utils = Utilities()
self._t = Transaction()
# Run program's main loop
self.run()
def getInput(self) -> str:
"""
Get input and catch potential errors.
Returns:
str: Input received from stdin.
"""
try:
inputs = input()
except EOFError:
exit(84)
except KeyboardInterrupt:
Logger("\nexit.", logType="INFO")
exit(0)
else:
return inputs
def initSettings(self, inputs: list) -> None:
"""
Parse the input passed as argument
and initialise corresponding class's attribute.
Args:
inputs (list): Input to be parsed.
"""
if inputs[globals.VARIABLE] == "candle_format":
# Candle format is set by default, don't need to stock it.
return
if self._utils.isInt(inputs[globals.VALUE]):
value = int(inputs[globals.VALUE])
elif self._utils.isFloat(inputs[globals.VALUE]):
value = float(inputs[globals.VALUE])
else:
value = inputs[globals.VALUE]
if inputs[globals.VARIABLE] == "initial_stack":
self._stack._USDT = value
if inputs[globals.VARIABLE] == "candles_given":
self._t._period = int(value / 3)
self._settings[inputs[globals.VARIABLE]] = value
def fetchCommand(self, command: list) -> None:
"""
Fetch the right command depending of the input and execute it.
Args:
command (list): the command to be checked and executed.
"""
if command[0] == "settings" and len(command) is 3:
self.initSettings(command)
elif command[0] == "update" and len(command) == 4:
if f"{command[1]} {command[2]}" == "game next_candles":
newCandle = Candle(command[3])
if newCandle._state == globals.VALID:
self._candles.append(newCandle)
self._t.update_lastClosePrices(newCandle)
elif f"{command[1]} {command[2]}" == "game stacks":
self._stack.update_s(command[3])
elif f"{command[0]} {command[1]}" == "action order":
# Server is waiting for a move. (`sell`|`buy`|`pass`)
self._t.strategy(candles=self._candles, stack=self._stack)
else:
Logger("Unrecognized command.")
def run(self) -> None:
"""
Main loop of the Trade project.
"""
while (True):
inputCommand = self.getInput()
# If no input: restart loop
if len(inputCommand) is 0:
continue
inputParsed = inputCommand.split()
# Fetch to the right command
self.fetchCommand(inputParsed)