You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
657 B
Python
23 lines
657 B
Python
import json
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from asteroid import Asteroid
|
|
|
|
class TestAsteroid(unittest.TestCase):
|
|
def setUp(self):
|
|
self.asteroid = Asteroid(2099942)
|
|
|
|
def mocked_get_data(self):
|
|
with open("apophis_fixture.txt") as f:
|
|
return json.loads(f.read())
|
|
|
|
@patch("asteroid.Asteroid.get_data", mocked_get_data)
|
|
def test_name(self):
|
|
self.assertEqual(
|
|
self.asteroid.name, "99942 Apophis (2004 MN4)"
|
|
)
|
|
|
|
@patch("asteroid.Asteroid.get_data", mocked_get_data)
|
|
def test_diameter(self):
|
|
self.assertEqual(self.asteroid.diameter, 682) |