Skip to content

Commit 6e8a286

Browse files
committed
Include new tests in 3D FFB tests
1 parent 848b9f4 commit 6e8a286

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/test_FFBbasis3D.py

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

66
from aspire.basis import FFBBasis3D
77

8+
from ._basis_util import Steerable3DMixin
9+
810
DATA_DIR = os.path.join(os.path.dirname(__file__), "saved_test_data")
911

1012

11-
class FFBBasis3DTestCase(TestCase):
13+
class FFBBasis3DTestCase(TestCase, Steerable3DMixin):
1214
def setUp(self):
15+
self.L = 8
1316
self.dtype = np.float32
14-
self.basis = FFBBasis3D((8, 8, 8), dtype=self.dtype)
17+
self.basis = FFBBasis3D((self.L, self.L, self.L), dtype=self.dtype)
18+
self.seed = 9161341
1519

1620
def tearDown(self):
1721
pass

0 commit comments

Comments
 (0)