def setUp(self):
# create a colon batch with 2 samples
self.test_batch = ColonMuscleBatch()
self.test_batch.create_parameters(2)
# artificially add 10 fake "reflectances" to this batch
# at 10 fake "wavelengths"
WAVELENGHTS = np.linspace(450, 720, 10)
reflectance1 = np.arange(0, 30, 3)
reflectance2 = np.arange(30, 60, 3)
for w in WAVELENGHTS:
self.test_batch.df["reflectances", w] = np.NAN
for r1, r2, w in zip(reflectance1, reflectance2, WAVELENGHTS):
self.test_batch.df["reflectances", w][0] = r1
self.test_batch.df["reflectances", w][1] = r2
# shortcut to dataframe that we are interested in:
self.df = self.test_batch.df
def create_batch_to_simulate(self):
return ColonMuscleBatch()
