class TestBanner:
url = 'localhost/banner1.jpg'
shows = 10
category1 = 'category1'
category2 = 'category2'
categories = [category1, category2]
def setup(self):
self.banner = Banner(TestBanner.url, TestBanner.shows,
TestBanner.categories)
def test_init(self):
assert self.banner.url == TestBanner.url
assert self.banner.shows == TestBanner.shows
assert self.banner.categories == TestBanner.categories
def test_str(self):
assert f'{self.banner.url}: {self.banner.shows}' == str(self.banner)
def test_show(self):
assert self.banner.show() == TestBanner.shows - 1
assert self.banner.show() == TestBanner.shows - 2
