import unittest

from swiftstory.Cards import Cards


class TestCards(unittest.TestCase):
    def test_get_cards_fr(self):
        self.assertTrue(Cards.get_white_cards("fr"))
        self.assertTrue(Cards.get_black_cards("fr"))

    def test_get_cards_en(self):
        self.assertTrue(Cards.get_white_cards("en"))
        self.assertTrue(Cards.get_black_cards("en"))

    def test_get_cards_unknown_language(self):
        with self.assertRaises(FileNotFoundError):
            Cards.get_white_cards("zz")
        with self.assertRaises(FileNotFoundError):
            Cards.get_black_cards("zz")