Python Trivia.get_category_id示例

    def test_trivia_get_category_id_valid(self):
        return_val = {
            "trivia_categories": [{
                "name": "cat1",
                "id": 1
            }, {
                "name": "cat2",
                "id": 2
            }, {
                "name": "cat3",
                "id": 3
            }]
        }

        with patch("trivia.Trivia.get_category_dict",
                   return_value=return_val) as mock_dict:
            self.assertEqual(Trivia.get_category_id("cat1"), 1)
            self.assertEqual(Trivia.get_category_id("cat2"), 2)
            self.assertEqual(Trivia.get_category_id("cat3"), 3)
            self.assertEqual(Trivia.get_category_id("cAT1"), 1)
            self.assertEqual(Trivia.get_category_id("CaT2"), 2)
            self.assertEqual(Trivia.get_category_id("cAt3"), 3)

            self.assertEqual(Trivia.get_category_id("Any"), 0)
            self.assertEqual(Trivia.get_category_id(None), 0)
            self.assertEqual(Trivia.get_category_id(""), 0)

示例#2

显示文件

文件： trivia_command_test.py 项目： mewmix/thedasbot

    def test_trivia_get_category_id_invalid(self):
        return_val = {"trivia_categories": [{"name": "science", "id": 1}]}

        with patch("trivia.Trivia.get_category_dict",
                   return_value=return_val) as mock_dict:
            self.assertEqual(Trivia.get_category_id("Science"), 1)
            self.assertRaises(BotError, Trivia.get_category_id, "bad")