diff --git a/tests/tests_models.py b/tests/tests_models.py index fdc18e0..0a60564 100644 --- a/tests/tests_models.py +++ b/tests/tests_models.py @@ -163,7 +163,9 @@ def test_item_perfect_shot(self): content=load_resource("content_item_perfect_shot.txt")) item = models.Item.from_article(article) self.assertIsInstance(item, models.Item) - self.assertEqual(11, len(item.attributes)) + self.assertEqual(12, len(item.attributes)) + self.assertIn("perfect_shot", item.attributes_dict) + self.assertIn("perfect_shot_range", item.attributes_dict) def test_key(self): article = Article(1, "Key 3940", timestamp="2018-08-20T04:33:15Z",