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)