def METHOD_NAME(self):
    expected_element_types = {
        'someString': 'STRING',
        'someDateTimeElement': 'DATETIME',
        'someOtherString': 'STRING',
        'someFloat': 'FLOAT64'
    }
    root = self.types_schema.elements['document']
    for element in root:
        self.assertEqual(xsd2bq.lookup_element_type(element.type),
                         expected_element_types[element.name])