From 5498c5e77c1ee8d2d2f8ddd155ff27a3665fb5d9 Mon Sep 17 00:00:00 2001 From: Joran Honig Date: Wed, 15 Aug 2018 12:06:01 +0200 Subject: [PATCH] Test fixes --- tests/taint_result_test.py | 2 +- tests/taint_runner_test.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/taint_result_test.py b/tests/taint_result_test.py index 1b2666e6..65909cbc 100644 --- a/tests/taint_result_test.py +++ b/tests/taint_result_test.py @@ -1,5 +1,5 @@ from mythril.laser.ethereum.taint_analysis import * -from mythril.laser.ethereum.svm import GlobalState +from mythril.laser.ethereum.state import GlobalState def test_result_state(): diff --git a/tests/taint_runner_test.py b/tests/taint_runner_test.py index 03666d2c..d52893c7 100644 --- a/tests/taint_runner_test.py +++ b/tests/taint_runner_test.py @@ -2,8 +2,8 @@ import mock import pytest from pytest_mock import mocker from mythril.laser.ethereum.taint_analysis import * -from mythril.laser.ethereum.svm import GlobalState, Node, Edge, LaserEVM -from mythril.laser.ethereum.state import MachineState, Account, Environment +from mythril.laser.ethereum.cfg import Node, Edge +from mythril.laser.ethereum.state import MachineState, Account, Environment, GlobalState def test_execute_state(mocker):