import unittest
class Sources: ''' News Sources class to define news source objects '''
def __init__(self,id,name,category,description):
self.id = id
self.name = name
self.category = category
self.description = description
class TestSources(unittest.TestCase): ''' Test class that defines test cases for the Sources class behaviours.
Args:
unittest.TestCase: TestCase class that helps in creating test cases
'''
def setUp(self):
'''
Set up method to run before each test cases.
'''
self.new_source = Sources("abc-news","ABC-NEWS","sports","") # create a source object
def test_init(self):
'''
test_init test case to test if the object is initialized properly
'''
self.assertEqual(self.new_source.id,"abc-news")
self.assertEqual(self.new_source.name,"ABC-NEWS")
self.assertEqual(self.new_source.category,"sports")
self.assertEqual(self.new_source.description,"")
if name == 'main': unittest.main()