tests: add import pytest where required

tests/device_tests/test_msg_getaddress_segwit.py

@@ -14,6 +14,8 @@
 # You should have received a copy of the License along with this library.
 # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
 
+import pytest
+
 from trezorlib import btc, messages as proto
 from trezorlib.tools import parse_path
 

tests/device_tests/test_msg_getaddress_segwit_native.py

@@ -14,6 +14,8 @@
 # You should have received a copy of the License along with this library.
 # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
 
+import pytest
+
 from trezorlib import btc, messages as proto
 from trezorlib.tools import parse_path
 

tests/device_tests/test_msg_getpublickey.py

@@ -14,6 +14,8 @@
 # You should have received a copy of the License along with this library.
 # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
 
+import pytest
+
 from trezorlib import btc, ckd_public as bip32, messages as proto
 from trezorlib.tools import H_
 

tests/device_tests/test_msg_signmessage.py

@@ -16,6 +16,8 @@
 
 import base64
 
+import pytest
+
 from trezorlib import btc
 from trezorlib.tools import parse_path
 

tests/device_tests/test_msg_signmessage_segwit.py

@@ -16,6 +16,8 @@
 
 import base64
 
+import pytest
+
 from trezorlib import btc, messages as proto
 from trezorlib.tools import parse_path
 

tests/device_tests/test_msg_signmessage_segwit_native.py

@@ -16,6 +16,8 @@
 
 import base64
 
+import pytest
+
 from trezorlib import btc, messages as proto
 from trezorlib.tools import parse_path
 

tests/device_tests/test_msg_signtx_dash.py

@@ -14,6 +14,8 @@
 # You should have received a copy of the License along with this library.
 # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
 
+import pytest
+
 from trezorlib import btc, messages as proto
 from trezorlib.tools import parse_path
 

tests/device_tests/test_msg_signtx_grs.py

@@ -14,6 +14,8 @@
 # You should have received a copy of the License along with this library.
 # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
 
+import pytest
+
 from trezorlib import btc, messages as proto
 from trezorlib.tools import parse_path
 

tests/device_tests/test_msg_verifymessage.py

@@ -16,6 +16,8 @@
 
 import base64
 
+import pytest
+
 from trezorlib import btc
 
 from .common import TrezorTest