diff --git a/tests/unit/test_account.py b/tests/unit/test_account.py index 347c0f843..f99ea60b7 100644 --- a/tests/unit/test_account.py +++ b/tests/unit/test_account.py @@ -1,6 +1,6 @@ from binascii import hexlify -from orchstr8.testcase import AsyncioTestCase +from torba.testing import AsyncioTestCase from torba.coin.bitcoinsegwit import MainNetLedger as ledger_class from torba.baseaccount import HierarchicalDeterministic, SingleKey diff --git a/tests/unit/test_bip32.py b/tests/unit/test_bip32.py index 1d3229f09..7ef4d74fa 100644 --- a/tests/unit/test_bip32.py +++ b/tests/unit/test_bip32.py @@ -1,6 +1,6 @@ from binascii import unhexlify, hexlify -from orchstr8.testcase import AsyncioTestCase +from torba.testing import AsyncioTestCase from .key_fixtures import expected_ids, expected_privkeys, expected_hardened_privkeys from torba.bip32 import PubKey, PrivateKey, from_extended_key_string diff --git a/tests/unit/test_coinselection.py b/tests/unit/test_coinselection.py index fc54639db..196d7aed0 100644 --- a/tests/unit/test_coinselection.py +++ b/tests/unit/test_coinselection.py @@ -1,6 +1,6 @@ from types import GeneratorType -from orchstr8.testcase import AsyncioTestCase +from torba.testing import AsyncioTestCase from torba.coin.bitcoinsegwit import MainNetLedger as ledger_class from torba.coinselection import CoinSelector, MAXIMUM_TRIES diff --git a/tests/unit/test_database.py b/tests/unit/test_database.py index d6fcc564e..a55aba8a1 100644 --- a/tests/unit/test_database.py +++ b/tests/unit/test_database.py @@ -5,7 +5,7 @@ from torba.constants import COIN from torba.coin.bitcoinsegwit import MainNetLedger as ledger_class from torba.basedatabase import query, constraints_to_sql -from orchstr8.testcase import AsyncioTestCase +from torba.testing import AsyncioTestCase from .test_transaction import get_output, NULL_HASH diff --git a/tests/unit/test_headers.py b/tests/unit/test_headers.py index 46489e733..dacb7832e 100644 --- a/tests/unit/test_headers.py +++ b/tests/unit/test_headers.py @@ -1,7 +1,7 @@ import os from urllib.request import Request, urlopen -from orchstr8.testcase import AsyncioTestCase +from torba.testing import AsyncioTestCase from torba.coin.bitcoinsegwit import MainHeaders diff --git a/tests/unit/test_transaction.py b/tests/unit/test_transaction.py index b082b6035..3e2ee01a1 100644 --- a/tests/unit/test_transaction.py +++ b/tests/unit/test_transaction.py @@ -2,7 +2,7 @@ import unittest from binascii import hexlify, unhexlify from itertools import cycle -from orchstr8.testcase import AsyncioTestCase +from torba.testing import AsyncioTestCase from torba.coin.bitcoinsegwit import MainNetLedger as ledger_class from torba.wallet import Wallet diff --git a/tests/unit/test_wallet.py b/tests/unit/test_wallet.py index baf65505c..87e0d66fa 100644 --- a/tests/unit/test_wallet.py +++ b/tests/unit/test_wallet.py @@ -1,6 +1,6 @@ import tempfile -from orchstr8.testcase import AsyncioTestCase +from torba.testing import AsyncioTestCase from torba.coin.bitcoinsegwit import MainNetLedger as BTCLedger from torba.coin.bitcoincash import MainNetLedger as BCHLedger diff --git a/torba/testing/__init__.py b/torba/testing/__init__.py index 7b2ed3e52..7f73aff48 100644 --- a/torba/testing/__init__.py +++ b/torba/testing/__init__.py @@ -1 +1 @@ -from .testcase import IntegrationTestCase +from .testcase import IntegrationTestCase, AsyncioTestCase