如何將單個斷言語句與for循環結合使用

Question

import unittest
from is_month_of_year import is_Month_Of_Year

class MonthsTestCase(unittest.TestCase):
    
    def test_is_January_month(self):
        self.assertTrue(is_Month_Of_Year('January'))
    
    def test_is_February_month(self):
        self.assertTrue(is_Month_Of_Year('February'))

    def test_is_March_month(self):
        self.assertTrue(is_Month_Of_Year('March'))
    
    def test_is_April_month(self):
        self.assertTrue(is_Month_Of_Year('April'))

    def test_is_May_month(self):
        self.assertTrue(is_Month_Of_Year('May'))
    
    def test_is_June_month(self):
        self.assertTrue(is_Month_Of_Year('June'))

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

Answer 1

import unittest
from is_month_of_year import is_Month_Of_Year

class MonthsTestCase(unittest.TestCase):
    
    def test_is_January_month(self):
        self.assertTrue(is_Month_Of_Year('January'))
    
    def test_is_February_month(self):
        self.assertTrue(is_Month_Of_Year('February'))

    def test_is_March_month(self):
        self.assertTrue(is_Month_Of_Year('March'))
    
    def test_is_April_month(self):
        self.assertTrue(is_Month_Of_Year('April'))

    def test_is_May_month(self):
        self.assertTrue(is_Month_Of_Year('May'))
    
    def test_is_June_month(self):
        self.assertTrue(is_Month_Of_Year('June'))

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