def test_megaohm_parsing(self): unit_reg = UnitRegistry() expected_unit = unit_reg.megaohm input_str = u'MΩ' output_str, found_unit = parse_units_from_str(input_str) self.assertEqual(found_unit, expected_unit)
def test_weird_unit_parsing(self): unit_reg = UnitRegistry() expected_unit = unit_reg.megaohm input_str = u'mV ms\u22121' found_unit = parse_units_from_str(input_str) self.assertIsNone(found_unit)
def test_millivolt_parsing(self): unit_reg = UnitRegistry() expected_unit = unit_reg.millivolt input_str = 'mV' output_str, found_unit = parse_units_from_str(input_str) self.assertEqual(found_unit, expected_unit)
def get_units_from_table_header(header_str): (a, parens_str, comma_str) = resolve_table_header(header_str) if parens_str: matched_units = parse_units_from_str(parens_str) if matched_units: return parens_str elif comma_str: matched_units = parse_units_from_str(comma_str) if matched_units: return comma_str else: # split header using whitespace and check if any are units for e in reversed(a.split()): matched_units = parse_units_from_str(e) if matched_units: return e return None