Unit Tests

taylorjdawson · taylorjdawson · commit de0829b9a136 · 2017-06-24T15:10:30.000-07:00
- I added unit tests for the Token methods
 - Updated the .gitignore file to ignore the unittest generated .cache file
diff --git a/.gitignore b/.gitignore
@@ -4,3 +4,4 @@
 # Temporary data
 .ipynb_checkpoints/
 /examples/stats/.ipynb_checkpoints
+/.cache/
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_token.py b/tests/test_token.py
@@ -0,0 +1,18 @@
+import pytest
+
+from etherscan.tokens import Tokens
+
+DGD_TOKEN_SUPPLY = '1994946756800000'
+DGD_TOKEN_BALANCE = '212900000000000'
+AP_IKEY = 'YourAPIkey'
+TOKEN_NAME = 'DGD'
+ADDRESS = '0x4366ddc115d8cf213c564da36e64c8ebaa30cdbd'
+API_KEY = 'YourAPIkey'
+
+def test_get_token_supply():
+    api = Tokens(tokenname=TOKEN_NAME, api_key=(API_KEY))
+    assert (api.get_total_supply() == DGD_TOKEN_SUPPLY)
+
+def test_get_token_balance():
+    api = Tokens(tokenname=TOKEN_NAME, api_key=API_KEY)
+    assert(api.get_token_balance(ADDRESS) == DGD_TOKEN_BALANCE)
