Fixed tests to use new Token constructor.
This commit is contained in:
Geoff Taylor
parent
e3b0d65705
commit
a239a27512
|
|
@ -2,7 +2,7 @@
|
|||
"cells": [
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "intense-illinois",
|
||||
"id": "urban-density",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# ⚠ Warning\n",
|
||||
|
|
@ -16,7 +16,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "ceramic-smoke",
|
||||
"id": "enclosed-postage",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# 🥭 WalletBalancer\n",
|
||||
|
|
@ -31,7 +31,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "adequate-costume",
|
||||
"id": "constitutional-today",
|
||||
"metadata": {
|
||||
"jupyter": {
|
||||
"source_hidden": true
|
||||
|
|
@ -54,7 +54,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "special-nomination",
|
||||
"id": "given-boutique",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# Target Balances\n",
|
||||
|
|
@ -75,7 +75,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "unknown-relationship",
|
||||
"id": "grand-product",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"## TargetBalance class\n",
|
||||
|
|
@ -86,7 +86,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "mighty-stock",
|
||||
"id": "typical-international",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -105,7 +105,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "earlier-internship",
|
||||
"id": "monthly-railway",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"## FixedTargetBalance class\n",
|
||||
|
|
@ -116,7 +116,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "electronic-cycle",
|
||||
"id": "trained-clear",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -134,7 +134,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "muslim-priority",
|
||||
"id": "front-gather",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"## PercentageTargetBalance\n",
|
||||
|
|
@ -150,7 +150,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "potential-mustang",
|
||||
"id": "complete-cleaning",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -170,7 +170,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "assigned-mississippi",
|
||||
"id": "formed-guest",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"## TargetBalanceParser class\n",
|
||||
|
|
@ -183,7 +183,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "hired-laundry",
|
||||
"id": "closed-verse",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -221,7 +221,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "private-colorado",
|
||||
"id": "legislative-evans",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# sort_changes_for_trades function\n",
|
||||
|
|
@ -234,7 +234,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "signal-reputation",
|
||||
"id": "fantastic-doctrine",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -244,7 +244,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "regular-array",
|
||||
"id": "alpha-intake",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# calculate_required_balance_changes function\n",
|
||||
|
|
@ -255,7 +255,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "minimal-kinase",
|
||||
"id": "superior-millennium",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -271,7 +271,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "round-appendix",
|
||||
"id": "undefined-forth",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# FilterSmallChanges class\n",
|
||||
|
|
@ -286,7 +286,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "european-ratio",
|
||||
"id": "proper-nicholas",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -315,7 +315,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "three-shareware",
|
||||
"id": "satisfactory-transparency",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# WalletBalancers\n",
|
||||
|
|
@ -331,7 +331,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "destroyed-validation",
|
||||
"id": "afraid-lightning",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"## WalletBalancer class\n",
|
||||
|
|
@ -342,7 +342,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "engaged-thunder",
|
||||
"id": "clear-workplace",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -354,7 +354,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "democratic-series",
|
||||
"id": "independent-electronics",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"## NullWalletBalancer class\n",
|
||||
|
|
@ -365,7 +365,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "wound-flour",
|
||||
"id": "under-pacific",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -376,7 +376,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "painful-stake",
|
||||
"id": "labeled-things",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"## LiveWalletBalancer class\n",
|
||||
|
|
@ -387,7 +387,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "south-ministry",
|
||||
"id": "spoken-bowling",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -454,7 +454,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "compatible-mustang",
|
||||
"id": "willing-trout",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# ✅ Testing"
|
||||
|
|
@ -463,7 +463,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "little-exposure",
|
||||
"id": "regulation-hurricane",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -471,9 +471,9 @@
|
|||
" log_level = logging.getLogger().level\n",
|
||||
" try:\n",
|
||||
" logging.getLogger().setLevel(logging.CRITICAL)\n",
|
||||
" eth = Token(\"ETH\", PublicKey(\"2FPyTwcZLUg1MDrwsyoP4D6s1tM7hAkHYRjkNb5w6Pxk\"), Decimal(6))\n",
|
||||
" btc = Token(\"BTC\", PublicKey(\"9n4nbM75f5Ui33ZbPYXn59EwSgE8CGsHtAeTH5YFeJ9E\"), Decimal(6))\n",
|
||||
" usdt = Token(\"USDT\", PublicKey(\"Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB\"), Decimal(6))\n",
|
||||
" eth = Token(\"ETH\", \"Wrapped Ethereum (Sollet)\", PublicKey(\"2FPyTwcZLUg1MDrwsyoP4D6s1tM7hAkHYRjkNb5w6Pxk\"), Decimal(6))\n",
|
||||
" btc = Token(\"BTC\", \"Wrapped Bitcoin (Sollet)\", PublicKey(\"9n4nbM75f5Ui33ZbPYXn59EwSgE8CGsHtAeTH5YFeJ9E\"), Decimal(6))\n",
|
||||
" usdt = Token(\"USDT\", \"USDT\", PublicKey(\"Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB\"), Decimal(6))\n",
|
||||
" current_prices = [\n",
|
||||
" TokenValue(eth, Decimal(\"4000\")),\n",
|
||||
" TokenValue(btc, Decimal(\"60000\")),\n",
|
||||
|
|
@ -491,9 +491,9 @@
|
|||
"\n",
|
||||
" changes = calculate_required_balance_changes(current_balances, desired_balances)\n",
|
||||
"\n",
|
||||
" assert(changes[0].token.name == \"ETH\")\n",
|
||||
" assert(changes[0].token.symbol == \"ETH\")\n",
|
||||
" assert(changes[0].value == Decimal(\"0.5\"))\n",
|
||||
" assert(changes[1].token.name == \"BTC\")\n",
|
||||
" assert(changes[1].token.symbol == \"BTC\")\n",
|
||||
" assert(changes[1].value == Decimal(\"-0.1\"))\n",
|
||||
"\n",
|
||||
" parsed_balance_change = FixedTargetBalance(eth, Decimal(\"0.1\"))\n",
|
||||
|
|
@ -548,7 +548,7 @@
|
|||
},
|
||||
{
|
||||
"cell_type": "markdown",
|
||||
"id": "emotional-instruction",
|
||||
"id": "thrown-italian",
|
||||
"metadata": {},
|
||||
"source": [
|
||||
"# 🏃 Running\n",
|
||||
|
|
@ -559,7 +559,7 @@
|
|||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": null,
|
||||
"id": "adjusted-myrtle",
|
||||
"id": "given-midwest",
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
|
|
@ -584,9 +584,9 @@
|
|||
" resolved = target.resolve(price, Decimal(10000))\n",
|
||||
" desired_balances += [resolved]\n",
|
||||
"\n",
|
||||
" assert(desired_balances[0].token.name == \"ETH\")\n",
|
||||
" assert(desired_balances[0].token.symbol == \"ETH\")\n",
|
||||
" assert(desired_balances[0].value == Decimal(\"0.5\"))\n",
|
||||
" assert(desired_balances[1].token.name == \"BTC\")\n",
|
||||
" assert(desired_balances[1].token.symbol == \"BTC\")\n",
|
||||
" assert(desired_balances[1].value == Decimal(\"0.05\"))\n",
|
||||
"\n",
|
||||
" current_balances = [\n",
|
||||
|
|
@ -604,9 +604,9 @@
|
|||
" # 0.5 ETH and 0.05 BTC, we need to sell 0.1 ETH and buy 0.04 BTC. But we want to do the sell\n",
|
||||
" # first, to make sure we have the proper liquidity when it comes to buying.\n",
|
||||
" sorted_changes = sort_changes_for_trades(changes)\n",
|
||||
" assert(sorted_changes[0].token.name == \"ETH\")\n",
|
||||
" assert(sorted_changes[0].token.symbol == \"ETH\")\n",
|
||||
" assert(sorted_changes[0].value == Decimal(\"-0.1\"))\n",
|
||||
" assert(sorted_changes[1].token.name == \"BTC\")\n",
|
||||
" assert(sorted_changes[1].token.symbol == \"BTC\")\n",
|
||||
" assert(sorted_changes[1].value == Decimal(\"0.04\"))\n"
|
||||
]
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue