def test_get_field_correspondence_list_4(self):
data = [[1, 'asdf', 'a0001']]
A = pd.DataFrame(data)
A.columns = ['ID', 'Title', 'isbn']
A_key = 'ID'
B = pd.DataFrame(data)
B.columns = ['Id', 'title', 'ISBN']
B_key = 'Id'
attr_corres = []
db._get_field_correspondence_list(A, B, A_key, B_key, attr_corres)
def test_get_field_correspondence_list_4(self):
data = [[1, 'asdf', 'a0001']]
A = pd.DataFrame(data)
A.columns = ['ID', 'Title', 'isbn']
A_key = 'ID'
B = pd.DataFrame(data)
B.columns = ['Id', 'title', 'ISBN']
B_key = 'Id'
attr_corres = []
db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
def test_get_field_correspondence_list_1(self):
A = read_csv_metadata(path_a, key='ID')
B = read_csv_metadata(path_b, key='ID')
A_key = em.get_key(A)
B_key = em.get_key(B)
expected_list = [('ID', 'ID'), ('name', 'name'),
('birth_year', 'birth_year'),
('hourly_wage', 'hourly_wage'),
('address', 'address'), ('zipcode', 'zipcode')]
attr_corres = None
corres_list = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
self.assertEqual(corres_list, expected_list)
attr_corres = []
corres_list = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
self.assertEqual(corres_list, expected_list)
def test_get_field_correspondence_list_1(self):
A = read_csv_metadata(path_a, key='ID')
B = read_csv_metadata(path_b, key='ID')
A_key = em.get_key(A)
B_key = em.get_key(B)
expected_list = [('ID', 'ID'), ('name', 'name'),
('birth_year', 'birth_year'),
('hourly_wage', 'hourly_wage'),
('address', 'address'),
('zipcode', 'zipcode')]
attr_corres = None
corres_list = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
self.assertEqual(corres_list, expected_list)
attr_corres = []
corres_list = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
self.assertEqual(corres_list, expected_list)
def test_get_field_correspondence_list_3(self):
data = [[1, 'asdf', 'a0001']]
A = pd.DataFrame(data)
A.columns = ['Id', 'Title', 'ISBN']
A_key = 'Id'
B = pd.DataFrame(data)
B.columns = ['Id', 'title', 'ISBN']
B_key = 'Id'
attr_corres = []
corres_list = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
expected_list = [('Id', 'Id'), ('ISBN', 'ISBN')]
self.assertEqual(corres_list, expected_list)
def test_get_field_correspondence_list_3(self):
data = [[1, 'asdf', 'a0001']]
A = pd.DataFrame(data)
A.columns = ['Id', 'Title', 'ISBN']
A_key = 'Id'
B = pd.DataFrame(data)
B.columns = ['Id', 'title', 'ISBN']
B_key = 'Id'
attr_corres = []
corres_list = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
expected_list = [('Id', 'Id'), ('ISBN', 'ISBN')]
self.assertEqual(corres_list, expected_list)
def test_get_field_correspondence_list_5(self):
A = pd.DataFrame([[0, 'A', 0.11, 'ASDF']])
A.columns = ['ID', 'name', 'price', 'desc']
em.set_key(A, 'ID')
A_key = em.get_key(A)
B = pd.DataFrame([['B', 'B001', 'ASDF', 0.111]])
B.columns = ['item_name', 'item_id', 'item_desc', 'item_price']
em.set_key(B, 'item_id')
B_key = em.get_key(B)
attr_corres = [('name', 'item_name'), ('price', 'item_price')]
actual_attr_corres = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
expected_attr_corres = [('name', 'item_name'), ('price', 'item_price'),
('ID', 'item_id')]
self.assertEqual(expected_attr_corres, actual_attr_corres)
def test_get_field_correspondence_list_5(self):
A = pd.DataFrame([[0, 'A', 0.11, 'ASDF']])
A.columns = ['ID', 'name', 'price', 'desc']
em.set_key(A, 'ID')
A_key = em.get_key(A)
B = pd.DataFrame([['B', 'B001', 'ASDF', 0.111]])
B.columns = ['item_name', 'item_id', 'item_desc', 'item_price']
em.set_key(B, 'item_id')
B_key = em.get_key(B)
attr_corres = [('name', 'item_name'),
('price', 'item_price')]
actual_attr_corres = db._get_field_correspondence_list(
A, B, A_key, B_key, attr_corres)
expected_attr_corres = [('name', 'item_name'),
('price', 'item_price'),
('ID', 'item_id')]
self.assertEqual(expected_attr_corres, actual_attr_corres)
