def test_yaml(self):
cards = dominion.load_cards()
self.assertGreater(len(cards), 0)
self.assertIn('Sets', cards)
self.assertNotIn('test_yaml', cards)
cards = dominion.load_cards('kingdom_builder/test_decks/test_yaml.yml')
self.assertIn('test_yaml', cards)
def test_flattening(self):
flattened = dominion.flatten_cards(self.card_data)
self.assertGreater(len(flattened), 0)
self.assertEquals(flattened[0].set, 'SetA')
self.card_data['Sets'][0]['name'] = 'SetC'
flattened = dominion.flatten_cards(self.card_data)
self.assertEquals(flattened[0].set, 'SetC')
cards = dominion.load_cards('kingdom_builder/test_decks/test_deck.yml')
flattened = dominion.flatten_cards(cards)
self.assertEquals(flattened[1].set, 'FileSetB')
