def test_planet_to_imperial(self):
p = Planet(id=1,
planet_id=2,
name='Test Planet',
diameter=123456,
rotation_period=1,
orbital_period=365,
gravity='1',
population=6000000,
climate='tundra,desert,rainforest',
terrain='flat',
surface_water=8)
imperial_dict = p.to_imperial()
# 123456*0.621371 = 76711
self.assertEqual(imperial_dict['diameter'], 76711)
def get_by_id_imperial(request, id):
try:
planet = Planet.objects.get(pk=id)
except Planet.DoesNotExist as dne:
success, response = get_planet_by_id(id)
planet = Planet(planet_id=id,
name=response['name'],
diameter=int(response['diameter']),
rotation_period=int(response['rotation_period']),
orbital_period=int(response['orbital_period']),
gravity=response['gravity'],
population=int(response['population']),
climate=response['climate'],
terrain=response['terrain'],
surface_water=int(response['surface_water']))
planet.save()
return JsonResponse(planet.to_imperial(), status=200)
