def test_user_tv(self):
john = User("John")
command = "tv_mood"
john.tv_switch(command)
self.assertEquals(john.name, "John")
self.assertEquals(john.tv_switch(command), TV.tv_display(TV))
def test_screen(self):
tv = TV
tv.tv_display(tv)
tv.dvd_display(tv)
self.assertEquals(tv.tv_display(tv), TV.tv_display(TV))
self.assertEquals(tv.dvd_display(tv), TV.dvd_display(TV))
def test_screen(self):
remote = Remote
remote.display_channel(TV)
remote.display_dvd(TV)
self.assertEquals(remote.display_channel(TV), TV.tv_display())
self.assertEquals(remote.display_channel(TV), TV.dvd_display())
