def test_valid_units(self): self.assertEqual('cm', get_unit('1cm')) self.assertEqual('mm', get_unit('3.1415mm')) self.assertEqual('%', get_unit('300%'))
def test_number(self): self.assertEqual(None, get_unit(1)) self.assertEqual(None, get_unit(1.0))