From 8c71aed2db970658d6b3e159cff26f0e0e09abeb Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Thu, 15 Jul 2021 10:42:21 -0400 Subject: [PATCH] Fix tests as well --- tests/__init__.py | 0 tests/liquidation_test.py | 7 ++++--- tests/tokenflow_test.py | 6 ++++-- 3 files changed, 8 insertions(+), 5 deletions(-) create mode 100644 tests/__init__.py diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/liquidation_test.py b/tests/liquidation_test.py index 8bc43f1..0771c2b 100644 --- a/tests/liquidation_test.py +++ b/tests/liquidation_test.py @@ -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() \ No newline at end of file + unittest.main() diff --git a/tests/tokenflow_test.py b/tests/tokenflow_test.py index 529bb35..7a13ba1 100644 --- a/tests/tokenflow_test.py +++ b/tests/tokenflow_test.py @@ -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() \ No newline at end of file + unittest.main()