def test_search_engine_1():
test = SearchEngine('test_dir1')
# Tests small number of words
assert_equals('test_dir1', test._dir)
assert_equals(5, test._num_docs)
assert_equals(0, test._calculate_idf('Cat'))
assert_equals(.223, test._calculate_idf('I'))
assert_equals(1.609, test._calculate_idf('dogs'))
assert_equals(['doc3.txt'], test.search('dogs'))
assert_equals(['doc3.txt'], test.search('dogs'))
assert_equals(['doc4.txt', 'doc1.txt'], test.search('eat'))
assert_equals(['doc4.txt', 'doc3.txt', 'doc1.txt'],
test.search('eat dogs'))
def test_read_function(url):
"""
Test method for read method in read_data class with given
url, does not crash if it passed
"""
web_data = DataReader(url)
data, columns = web_data.read()
assert_equals(True, len(data) > 0)
assert_equals(True, len(columns) > 0)
assert_equals(252, len(data))
assert_equals(15, len(columns))
assert_equals(23, data.loc[0, 'Age (years)'])
assert_equals('Weight (lbs)', columns[3])
return data, columns
def test_get_dname():
"""
Test the function get_dname from the SearchEngine class
"""
assert_equals('test', se.get_dname())
def test_get_filename():
"""
Test the function get_fname from the document class
"""
assert_equals('dog.txt', doc.get_filename())
assert_equals('abcd.txt', doc2.get_filename())
def test_clean_map_data(data_2019, geo_df):
"""
Takes data_2019 and geo_df as parameters
Checks the number of columns in the cleaned data
"""
assert_equals(4, len(cleanData.clean_map_data(data_2019, geo_df).columns))
def test_question_3(data):
"""
provide different cases to test methods in question_3
program crashes if don't meet the expection result
"""
assert_equals(("Spectre", 23995), q3.richest_hero(data))
assert_equals(("Earthshaker", 8598), q3.poorest_hero(data))
assert_equals(("Spectre", 18), q3.killer_hero(data))
assert_equals(("Queen of Pain", 15), q3.burden_hero(data))
assert_equals(("Clockwerk", 3), q3.escape_hero(data))
assert_equals(("Alchemist", 2), q3.most_popular_hero(data))
assert_equals(("Abaddon", 1), q3.least_popular_hero(data))
assert_equals(("Spectre", 31122), q3.damage_hero(data))