Skip to content

Commit d7e0199

Browse files
committed
Include new tests in 3D FFB tests
1 parent 8b3643d commit d7e0199

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/test_FFBbasis3D.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,17 @@
55

66
from aspire.basis import FFBBasis3D
77

8-
from ._basis_util import UniversalBasisMixin
8+
from ._basis_util import Steerable3DMixin, UniversalBasisMixin
99

1010
DATA_DIR = os.path.join(os.path.dirname(__file__), "saved_test_data")
1111

1212

13-
class FFBBasis3DTestCase(TestCase, UniversalBasisMixin):
13+
class FFBBasis3DTestCase(TestCase, Steerable3DMixin, UniversalBasisMixin):
1414
def setUp(self):
1515
self.L = 8
1616
self.dtype = np.float32
1717
self.basis = FFBBasis3D((self.L, self.L, self.L), dtype=self.dtype)
18+
self.seed = 9161341
1819

1920
def tearDown(self):
2021
pass

0 commit comments

Comments
 (0)