class MockSize(object):
- def __init__(self, factor):
+ def __init__(self, factor, preemptible=False):
self.id = 'z{}.test'.format(factor)
self.name = 'test size '+self.id
self.ram = 128 * factor
self.price = float(factor)
self.extra = {}
self.real = self
- self.preemptable = False
+ self.preemptible = preemptible
def __eq__(self, other):
return self.id == other.id