def test_table(self):
schema = init_schema(self.sample_input_4_class)
with sample_db(self.sample_input_4_class,
self.factor_map_treated_sex) as job:
self.assertEquals(np.shape(job.input.table), (1000, 16))
def test_table(self):
schema = init_schema(self.sample_input_4_class)
with sample_db(self.sample_input_4_class,
self.factor_map_treated_sex) as job:
self.assertEquals(np.shape(job.input.table), (1000, 16))
def test_model_to_layout(self):
with sample_db(self.sample_input_4_class,
self.factor_map_treated_sex) as job:
# One class
model = Model(job.schema, 'treated')
self.assertEquals(
model.layout,
[[0, 1, 2, 3, 4, 5, 6, 7], [8, 9, 10, 11, 12, 13, 14, 15]])
# No classes
model = Model(job.schema, '')
self.assertEquals(
model.layout,
[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]])
def test_model_to_layout(self):
with sample_db(self.sample_input_4_class,
self.factor_map_treated_sex) as job:
# One class
model = Model(job.schema, 'treated')
self.assertEquals(
model.layout,
[[0, 1, 2, 3, 4, 5, 6, 7],
[8, 9, 10, 11, 12, 13, 14, 15]])
# No classes
model = Model(job.schema, '')
self.assertEquals(
model.layout,
[[0, 1, 2, 3, 4, 5, 6, 7,
8, 9, 10, 11, 12, 13, 14, 15]])
