def test_get_indexes_num_elems(self): self.config.attributes.observables.types = { 'gender': CommonTypes.any.value } self.config.attributes_dict = load_attributes_dict(self.config) indexes = get_indexes(self.config) self.assertEqual(len(indexes), 729)
def initialize(self): self.excluded = load_excluded(self) self.annotations_dict = load_annotations_dict(self) subset_annotations(self) self.attributes_dict = load_attributes_dict(self) self.attributes_indexes = get_indexes(self) subset_attributes(self) self.cells_dict = load_cells_dict(self) subset_cells(self)
def test_load_attributes_dict_age_range(self): attributes_dict = load_attributes_dict(self.config) self.assertEqual( max(attributes_dict['age']) - min(attributes_dict['age']), 80)
def test_load_attributes_dict_num_elems(self): attributes_dict = load_attributes_dict(self.config) self.assertEqual(len(attributes_dict['age']), 729)
def test_load_attributes_dict_num_keys(self): attributes_dict = load_attributes_dict(self.config) self.assertEqual(len(list(attributes_dict.keys())), 2)
def test_get_indexes_num_m(self): self.config.attributes.observables.types = {'gender': 'M'} self.config.attributes_dict = load_attributes_dict(self.config) indexes = get_indexes(self.config) self.assertEqual(len(indexes), 341)
def test_pass_indexes_num_f(self): self.config.attributes.observables.types = {'gender': 'F'} self.config.attributes_dict = load_attributes_dict(self.config) indexes = pass_indexes(self.config, 'gender', 'F', 'any') self.assertEqual(len(indexes), 388)
def test_pass_indexes_num_m(self): self.config.attributes.observables.types = {'gender': 'M'} self.config.attributes_dict = load_attributes_dict(self.config) indexes = pass_indexes(self.config, 'gender', 'M', CommonTypes.any.value) self.assertEqual(len(indexes), 341)