def test_memory(self): mega = Memory(1, "Mb") self.assertEqual(mega.to("byte"), 1024**2) self.assertEqual(mega, Memory(1, "mb")) same_mega = Memory.from_string("1Mb") self.assertEqual(same_mega.unit_type, "memory") other_mega = Memory.from_string("+1.0 mb") self.assertEqual(mega, other_mega)
def any2mb(s): """Convert string or number to memory in megabytes.""" if is_string(s): return int(Memory.from_string(s).to("Mb")) else: return int(s)