def test_read_csv(self):
"""Reads csv file"""
dognames = student_submission.read_csv('./dognames.csv')
number_of_lines = len(dognames)
self.assertEqual(number_of_lines, 6980, 'Unexpected number of lines')
correct_type = all(
type(row) is collections.OrderedDict for row in dognames)
self.assertTrue(correct_type,
'The type of the list elements seems incorrect')
has_dogname_header = all('HUNDENAME' in row for row in dognames)
has_birthyear_header = all('GEBURTSJAHR_HUND' in row
for row in dognames)
has_sex_header = all('GESCHLECHT_HUND' in row for row in dognames)
self.assertTrue(
has_dogname_header,
'There seems to be something wrong with the "HUNDENAME" column')
self.assertTrue(
has_birthyear_header,
'There seems to be something wrong with the "GEBURTSJAHR_HUND" column'
)
self.assertTrue(
has_sex_header,
'There seems to be something wrong with the "GESCHLECHT_HUND" column'
)
only_three_keys = all(len(row.keys()) == 3 for row in dognames)
self.assertTrue(
only_three_keys,
'There seems to be an incorrect number of columns in your list')
def test_filter_dognames(self):
"""Test if filter_dognames works"""
dognames = student_submission.read_csv('./dognames.csv')
unfiltered = student_submission.filter_dognames(dognames)
self.assertTrue(isinstance(unfiltered, list),
'The return type of filter_dognames seems wrong')
self.assertTrue(
all(
isinstance(unfiltered[i], collections.OrderedDict)
for i in range(len(dognames))))
self.assertEqual(
len(student_submission.filter_dognames(dognames, year=2004)), 425,
'Your implementation seems wrong')
self.assertEqual(
len(student_submission.filter_dognames(dognames, sex='m')), 3431,
'Your implementation seems wrong')
self.assertEqual(
len(
student_submission.filter_dognames(dognames,
year=2004,
sex='m')), 217,
'Your implementation seems wrong')
self.assertEqual(
student_submission.filter_dognames(dognames, year=2004,
sex='m')[43]['HUNDENAME'],
"Charly", 'Your implementation seems wrong')
def test_dognames_count(self):
"""Test if dognames_count works"""
dognames = student_submission.read_csv('./dognames.csv')
count = student_submission.dognames_count(dognames)
self.assertTrue(isinstance(count, dict),
'The return type of dognames_count seems wrong')
self.assertEqual(count['Rocky'], 10, 'The counting seems to be wrong')
def test_is_valid_row(self):
"""Test if is_valid_row works"""
dognames = student_submission.read_csv('./dognames.csv')
self.assertTrue(
student_submission.is_valid_row(dognames[999], year=2007, sex='m'),
'Your implementation seems wrong')
self.assertFalse(
student_submission.is_valid_row(dognames[999], year=2006, sex='w'),
'Your implementation seems wrong')
def test_top_n_dognames(self):
"""Test if top_n_dognames works"""
dognames = student_submission.read_csv('./dognames.csv')
top_1 = student_submission.top_n_dognames(dognames, 1)
self.assertTrue(isinstance(top_1, list),
'The return type of top_n_dognames seems wrong')
self.assertEqual(top_1[0][0], 'Luna',
'The top used dog name seems wrong')
def test_filter_dognames(self):
"""Test if filter_dognames works"""
dognames = student_submission.read_csv('./dognames.csv')
unfiltered = student_submission.filter_dognames(dognames)
self.assertTrue(isinstance(unfiltered, list),
'The return type of filter_dognames seems wrong')
self.assertEqual(
len(student_submission.filter_dognames(dognames, year=2004)), 98,
'Your implementation seems wrong')
def test_preview_dognames(self):
"""Test if preview_dognames works"""
dognames = student_submission.read_csv('./dognames.csv')
preview = student_submission.preview_dognames(dognames, 1)
self.assertTrue(isinstance(preview, list),
'The return type of preview_dognames seems wrong')
preview_all = student_submission.preview_dognames(dognames, 1396)
self.assertEqual(len(preview_all), 1396,
'The length of the result list seems wrong')
all_strings = all(isinstance(name, str) for name in preview_all)
self.assertTrue(all_strings,
'The result list should contain only strings')
def test_top_n_dognames(self):
"""Test if top_n_dognames works"""
dognames = student_submission.read_csv('./dognames.csv')
top_1 = student_submission.top_n_dognames(dognames, 1)
self.assertTrue(isinstance(top_1, list),
'The return type of top_n_dognames seems wrong')
self.assertEqual(top_1[0][0], 'Luna',
'The top used dog name seems wrong')
top_100 = student_submission.top_n_dognames(dognames, 100)
self.assertEqual(len(top_100), 100,
'The length of the list seems wrong')
is_sorted = all(top_100[i][1] >= top_100[i + 1][1]
for i in range(len(top_100) - 1))
self.assertTrue(is_sorted,
'Your list does not seem to be sorted correctly')
def test_is_valid_row(self):
"""Test if is_valid_row works"""
dognames = student_submission.read_csv('./dognames.csv')
self.assertTrue(student_submission.is_valid_row(dognames[999]),
'Your implementation seems wrong')
self.assertTrue(
student_submission.is_valid_row(dognames[999], year=2010),
'Your implementation seems wrong')
self.assertTrue(
student_submission.is_valid_row(dognames[999], sex='m'),
'Your implementation seems wrong')
self.assertTrue(
student_submission.is_valid_row(dognames[999], year=2010, sex='m'),
'Your implementation seems wrong')
self.assertFalse(
student_submission.is_valid_row(dognames[999], year=2006, sex='m'),
'Your implementation seems wrong')
self.assertFalse(
student_submission.is_valid_row(dognames[999], year=2010, sex='w'),
'Your implementation seems wrong')
self.assertEqual(
sum(
student_submission.is_valid_row(dognames[i])
for i in range(len(dognames))), 6980,
'Your implementation seems wrong')
self.assertEqual(
sum(
student_submission.is_valid_row(dognames[i], sex='w')
for i in range(len(dognames))), 3549,
'Your implementation seems wrong')
self.assertEqual(
sum(
student_submission.is_valid_row(dognames[i], year=2000)
for i in range(len(dognames))), 118,
'Your implementation seems wrong')
def test_read_csv(self):
"""Reads csv file"""
dognames = student_submission.read_csv('./dognames.csv')
number_of_lines = len(dognames)
self.assertEqual(number_of_lines, 1396, 'Unexpected number of lines')