def test_surnames(self):
self.assertIn(
name_generator.get_surnames(1)[0],
name_generator.load_file('surnames'))
def test_names_both(self):
self.assertIn(
name_generator.get_names(1)[0],
name_generator.load_file('names_both'))
def test_names_male(self):
self.assertIn(
name_generator.get_male_names(1)[0],
name_generator.load_file('names_male'))
def test_load_testfile2(self):
result = name_generator.load_file('testfile2')
self.assertEqual(len(result), 1)
self.assertEqual(result, testfile2_content)