def test_display_data(self): ''' Testing if the data can be displayed. ''' self.new_data.add_password() test_data = UsersInfo(1, 1, "gmail.com", "kiriu") test_data.add_password() data_found = UsersInfo.display_data(1, 1) self.assertEqual(data_found.website, test_data.website)
def display_data(data, number): ''' Function that displays the user data ''' return UsersInfo.display_data(data, number)