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]])