Pythonでのunittestにおいて、辞書型のオブジェクトに期待するkeyが存在するかテストしたのでテスト方法をメモしておく。

テスト方法

辞書型オブジェクトに期待するkeyが存在するかテストするために、unittestのassertIn()を使用した。

assertIn(a,b)

assertIn(a,b)は、aがbに含まれているかテストするメソッド。これを使ったサンプルが以下となる。

import unittest

class TestSample(unittest.TestCase):
    def test_sample1(self):
        test_data = {
            "name": "ryota",
            "score": 63,
        }
        self.assertIn("score", test_data)

    def test_sample2(self):
        test_data = {"name": "ryota", "score": {"Math": 63, "English": 50}}
        self.assertIn("Math", test_data["score"])
        # これだとダメ
        # self.assertIn("Math", test_data) 

if __name__ == "__main__":
    unittest.main()

また、assertIn(a,b)の否定がassertNotIn(a,b)となる。