class SongTest (unittest.TestCase):
def setUp (self):
self.song1 = Song ("ABC", "abc")
def test_unlock_values (self):
new_values = [1, 2, 3, 4, 5]
self.song1.unlock_values (new_values)
new_list = [self.song1.arcade, self.song1.clear, self.song1.dance, self.song1.roulette, self.song1.song]
for i, item in enumerate (new_list):
self.assertEqual (item, new_values[i])
def test_reset (self):
self.song1.arcade = 500
self.song1.clear = 10
self.song1.dance = 5
self.song1.roulette = 10
self.song1.song = 10000
defaults = list (self.song1._defaults)
self.song1.reset ()
new_list = [self.song1.arcade, self.song1.clear, self.song1.dance, self.song1.roulette, self.song1.song]
for i, item in enumerate (new_list):
self.assertEqual (item, defaults[i])
def setUp (self):
self.song1 = Song ("ABC", "abc")
