Skip to content

Commit a311f9b

Browse files
committed
Add mixin to FFB 2D tests
1 parent abda13a commit a311f9b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_FFBbasis2D.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,15 +11,18 @@
1111
from aspire.utils.misc import grid_2d
1212
from aspire.volume import Volume
1313

14+
from ._basis_util import Steerable2DMixin
15+
1416
logger = logging.getLogger(__name__)
1517
DATA_DIR = os.path.join(os.path.dirname(__file__), "saved_test_data")
1618

1719

18-
class FFBBasis2DTestCase(TestCase):
20+
class FFBBasis2DTestCase(TestCase, Steerable2DMixin):
1921
def setUp(self):
2022
self.dtype = np.float32 # Required for convergence of this test
2123
self.L = 8
2224
self.basis = FFBBasis2D((self.L, self.L), dtype=self.dtype)
25+
self.seed = 9161341
2326

2427
def tearDown(self):
2528
pass

0 commit comments

Comments
 (0)