def test_raises_exception_when_attribute_is_missing(self):
scale = Scale(attribute_code='weight')
p = factories.create_product()
with self.assertRaises(ValueError):
scale.weigh_product(p)
def test_uses_default_weight_when_attribute_is_missing(self):
scale = Scale(attribute_code='weight', default_weight=0.5)
p = factories.create_product()
self.assertEqual(0.5, scale.weigh_product(p))
def test_weighs_uses_specified_attribute(self):
scale = Scale(attribute_code='weight')
p = factories.create_product(attributes={'weight': '1'})
self.assertEqual(1, scale.weigh_product(p))
def test_raises_exception_when_attribute_is_missing(self):
scale = Scale(attribute_code='weight')
p = factories.create_product()
with self.assertRaises(ValueError):
scale.weigh_product(p)
def test_uses_default_weight_when_attribute_is_missing(self):
scale = Scale(attribute_code='weight', default_weight=0.5)
p = factories.create_product()
self.assertEqual(0.5, scale.weigh_product(p))
def test_weighs_uses_specified_attribute(self):
scale = Scale(attribute_code='weight')
p = factories.create_product(attributes={'weight': '1'})
self.assertEqual(1, scale.weigh_product(p))
