diff --git a/test/functional/data/invalid_txs.py b/test/functional/data/invalid_txs.py index 3903bf9de4d..7a769922ebf 100644 --- a/test/functional/data/invalid_txs.py +++ b/test/functional/data/invalid_txs.py @@ -58,6 +58,7 @@ from test_framework.script import ( OP_SUBSTR, OP_XOR, ) +from test_framework.util import assert_equal from test_framework.script_util import ( MIN_PADDING, MIN_STANDARD_TX_NONWITNESS_SIZE, diff --git a/test/functional/feature_posix_fs_permissions.py b/test/functional/feature_posix_fs_permissions.py index e1884694b57..5fc82c26c42 100755 --- a/test/functional/feature_posix_fs_permissions.py +++ b/test/functional/feature_posix_fs_permissions.py @@ -9,6 +9,7 @@ import os import stat from test_framework.test_framework import BitcoinTestFramework +from test_framework.util import assert_equal class PosixFsPermissionsTest(BitcoinTestFramework): diff --git a/test/functional/feature_reindex_readonly.py b/test/functional/feature_reindex_readonly.py index 858a67566f9..6f088e4a39c 100755 --- a/test/functional/feature_reindex_readonly.py +++ b/test/functional/feature_reindex_readonly.py @@ -10,6 +10,7 @@ import os import stat import subprocess from test_framework.test_framework import BitcoinTestFramework +from test_framework.util import assert_equal class BlockstoreReindexTest(BitcoinTestFramework): diff --git a/test/functional/mining_getblocktemplate_longpoll.py b/test/functional/mining_getblocktemplate_longpoll.py index 421116965ce..db547971e05 100755 --- a/test/functional/mining_getblocktemplate_longpoll.py +++ b/test/functional/mining_getblocktemplate_longpoll.py @@ -8,7 +8,7 @@ import random import threading from test_framework.test_framework import BitcoinTestFramework -from test_framework.util import get_rpc_proxy +from test_framework.util import assert_equal, get_rpc_proxy from test_framework.wallet import MiniWallet diff --git a/test/functional/p2p_handshake.py b/test/functional/p2p_handshake.py index 774ab42980d..a05c072b316 100755 --- a/test/functional/p2p_handshake.py +++ b/test/functional/p2p_handshake.py @@ -9,7 +9,7 @@ import itertools import time from test_framework.test_framework import BitcoinTestFramework -from test_framework.util import assert_not_equal +from test_framework.util import assert_equal, assert_not_equal from test_framework.messages import ( NODE_NETWORK, NODE_NETWORK_LIMITED, diff --git a/test/functional/test_framework/crypto/ellswift.py b/test/functional/test_framework/crypto/ellswift.py index 2a2caa813ed..db48dd24682 100644 --- a/test/functional/test_framework/crypto/ellswift.py +++ b/test/functional/test_framework/crypto/ellswift.py @@ -13,6 +13,7 @@ import random import unittest from test_framework.crypto.secp256k1 import FE, G, GE +from test_framework.util import assert_equal # Precomputed constant square root of -3 (mod p). MINUS_3_SQRT = FE(-3).sqrt() diff --git a/test/functional/test_framework/crypto/secp256k1.py b/test/functional/test_framework/crypto/secp256k1.py index 479d046fa9b..b7193d46cd2 100644 --- a/test/functional/test_framework/crypto/secp256k1.py +++ b/test/functional/test_framework/crypto/secp256k1.py @@ -17,7 +17,7 @@ Exports: import unittest from hashlib import sha256 -from test_framework.util import assert_not_equal +from test_framework.util import assert_equal, assert_not_equal class FE: """Objects of this class represent elements of the field GF(2**256 - 2**32 - 977). diff --git a/test/functional/test_framework/key.py b/test/functional/test_framework/key.py index 6caafe8bcdb..1e963dd3e9e 100644 --- a/test/functional/test_framework/key.py +++ b/test/functional/test_framework/key.py @@ -14,7 +14,7 @@ import random import unittest from test_framework.crypto import secp256k1 -from test_framework.util import assert_not_equal, random_bitflip +from test_framework.util import assert_equal, assert_not_equal, random_bitflip # Point with no known discrete log. H_POINT = "50929b74c1a04954b78b4b6035e97a5e078a5a0f28ec96d547bfee9ace803ac0" diff --git a/test/functional/test_framework/psbt.py b/test/functional/test_framework/psbt.py index 8c552e420eb..5790d634e13 100644 --- a/test/functional/test_framework/psbt.py +++ b/test/functional/test_framework/psbt.py @@ -5,6 +5,7 @@ import base64 +from .util import assert_equal from .messages import ( CTransaction, deser_string, diff --git a/test/functional/wallet_disable.py b/test/functional/wallet_disable.py index dbcccd47784..46b92ad577f 100755 --- a/test/functional/wallet_disable.py +++ b/test/functional/wallet_disable.py @@ -9,7 +9,7 @@ """ from test_framework.test_framework import BitcoinTestFramework -from test_framework.util import assert_raises_rpc_error +from test_framework.util import assert_raises_rpc_error, assert_equal class DisableWalletTest (BitcoinTestFramework): def set_test_params(self):