def test_get_class_data(self): kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv') kb.add_data('..\\..\\test_data\\Test_Data_C.csv') print(kb.get_data()) print() print(kb.get_class_data(['A', 'C'])) print() print(kb.get_class_data(['B'])) print(kb.get_class_data(['Z'])) print(kb.get_class_data(['A', 'B', 'Z']))
def test_add_data(self): kb = KnowledgeBase(None, '..\\..\\test_data\\Test_Data_AB.csv') print(kb.get_data()) print() kb.add_data('..\\..\\test_data\\Test_Data_C.csv') print(kb.get_data())