Fix tests as well

This commit is contained in:
Luke Van Seters 2021-07-15 10:42:21 -04:00
parent 39a2038d95
commit 8c71aed2db
3 changed files with 8 additions and 5 deletions

0
tests/__init__.py Normal file
View File

View File

@ -1,6 +1,7 @@
import unittest
import inspector_compound
import inspector_aave
from mev_inspect import inspector_compound
from mev_inspect import inspector_aave
class TestLiquidations (unittest.TestCase):
def test_compound_liquidation(self):
@ -16,4 +17,4 @@ class TestLiquidations (unittest.TestCase):
if __name__ == '__main__':
unittest.main()
unittest.main()

View File

@ -1,5 +1,7 @@
import unittest
import tokenflow
from mev_inspect import tokenflow
class TestTokenFlow (unittest.TestCase):
def test_simple_arb(self):
@ -24,4 +26,4 @@ class TestTokenFlow (unittest.TestCase):
self.assertEqual(res['dollar_flows'], [0,0])
if __name__ == '__main__':
unittest.main()
unittest.main()