def test_storage_and_retrieval_of_measurement(self): arbitrary_default_unit='lb' arbitrary_value=124 arbitrary_measure=measure.Weight arbitrary_measurement=utils.get_measurement( arbitrary_measure, arbitrary_value, arbitrary_default_unit, ) instance = MeasurementTestModel.objects.create( measurement=arbitrary_measurement ) instance.save() retrieved = MeasurementTestModel.objects.all()[0] self.assertEqual( retrieved.measurement._default_unit, arbitrary_default_unit, ) self.assertEqual( retrieved.measurement.__class__, arbitrary_measure, ) self.assertAlmostEqual( retrieved.measurement.lb, arbitrary_value, )
def test_get_measurement(self): expected_measurement = measure.Volume(us_qt=34) actual_measurement = utils.get_measurement( measure.Volume, 34, 'us_qt', ) self.assertEqual( expected_measurement, actual_measurement, )