def test_patients_some_missing(self):
        """One patient, with the value present."""

        id_to_attributes = {
            'Tom': {
                'Pet': 'dog',
                'Car': 'NA',
                'Team': 'Leafs'
            },
            'Lily': {
                'Pet': 'dog',
                'Car': 'NA',
                'Team': 'Leafs'
            },
            'Jay': {
                'Pet': 'dog',
                'Car': 'BMW',
                'Team': 'Leafs'
            }
        }
        name = 'Car'
        expected = ['Tom', 'Lily']
        actual = tfs.patients_with_missing_values(id_to_attributes, name)

        msg = ERROR_MESSAGE.format(expected, actual)
        self.assertEqual(expected, actual, msg)
    def test_empty(self):
        """Empty dictionary."""

        id_to_attributes = {}
        name = 'xyz'
        expected = []
        actual = tfs.patients_with_missing_values(id_to_attributes, name)

        msg = ERROR_MESSAGE.format(expected, actual)
        self.assertEqual(expected, actual, msg)
Exemple #3
0
    'read_patients_dataset(a non-empty file)',
    'a non-empty Dict[str, Dict[str, str]]',
    actual)

########## build_value_to_ids
actual = tf.build_value_to_ids(d, 'Gender')
check_type_details(
    isinstance(actual, dict) and 'female' in actual and
    isinstance(actual['female'], list) and actual['female'] != [] and
    isinstance(actual['female'][0], str),
    'build_value_to_ids',
    'a non-empty Dict[str, List[str]] with a non-empty value',
    actual)

######### patients_with_missing_values
actual = tf.patients_with_missing_values(d, 'Age')
check_type_details(
    isinstance(actual, list) and actual != [] and isinstance(actual[0], str),
    'patients_with_missing_values',
    'a non-empty List[str]',
    actual)

######### similarity_score
check_function(tf.similarity_score, [d['tcga.5l.aat0'], p], float)

######### patient_similarities
actual = tf.patient_similarities(d, p)
check_type_details(
    isinstance(actual, dict) and 'tcga.5l.aat0' in actual and
    isinstance(actual['tcga.5l.aat0'], float),
    'patient_similarities',