def test_2D(self):
model = 1
phantom = TomoPhantom.get_ImageData(model, self.ig)
assert phantom.geometry.channels == 1
assert phantom.shape == (self.N, self.N)
def test_MC3D(self):
model = 100
ag = self.ag.copy()
ag.set_channels(5)
ig = ag.get_ImageGeometry()
phantom = TomoPhantom.get_ImageData(model, ig)
assert phantom.geometry.channels == ig.channels
assert phantom.shape == ig.shape
def test_MC2D_wrong_amount_channels(self):
model = 1
ag = self.ag.copy()
ag.set_channels(3)
ig = ag.get_ImageGeometry()
try:
phantom = TomoPhantom.get_ImageData(model, ig)
assert False
except ValueError as ve:
assert True
def test_get_model_num_channels(self):
model = 100
N = TomoPhantom.is_model_temporal(model, num_dims=2)
assert N >= 1
def test_is_model_temporal(self):
model = 100
assert TomoPhantom.is_model_temporal(model, num_dims=2)
def test_MC2D(self):
model = 100
self.ag.set_channels(3)
phantom = TomoPhantom.get_ImageData(model, self.ig)
assert phantom.geometry.channels == self.ig.channels
assert phantom.shape == self.ig.shape