def test_ab_block_candset_empty_output(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1, r_block_attr_1)
D = self.ab.block_candset(C, l_block_attr_3, r_block_attr_3)
assert_equal(len(D), 0)
assert_equal(sorted(D.columns), sorted(C.columns))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(D, 'fk_ltable'), mg.get_property(C, 'fk_ltable'))
assert_equal(mg.get_property(D, 'fk_rtable'), mg.get_property(C, 'fk_rtable'))
def test_ab_block_tables_wi_no_output_tuples(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_3, r_block_attr_3)
assert_equal(len(C), 0)
assert_equal(sorted(C.columns), sorted(['_id', 'ltable_' + l_key,
'rtable_' + r_key]))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(C, 'fk_ltable'), 'ltable_' + l_key)
assert_equal(mg.get_property(C, 'fk_rtable'), 'rtable_' + r_key)
def test_ab_block_tables_wi_empty_r_output_attrs(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1, r_block_attr_1, l_output_attrs, [])
s1 = ['_id', 'ltable_' + l_key, 'rtable_' + r_key]
s1 += ['ltable_' + x for x in l_output_attrs if x != l_key]
s1 = sorted(s1)
assert_equal(s1, sorted(C.columns))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(C, 'fk_ltable'), 'ltable_' + l_key)
assert_equal(mg.get_property(C, 'fk_rtable'), 'rtable_' + r_key)
def test_ab_block_tables_wi_no_output_tuples(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_3,
r_block_attr_3)
assert_equal(len(C), 0)
assert_equal(sorted(C.columns),
sorted(['_id', 'ltable_' + l_key, 'rtable_' + r_key]))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(C, 'fk_ltable'), 'ltable_' + l_key)
assert_equal(mg.get_property(C, 'fk_rtable'), 'rtable_' + r_key)
def test_ab_block_tables_wi_empty_r_output_attrs(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1,
r_block_attr_1, l_output_attrs, [])
s1 = ['_id', 'ltable_' + l_key, 'rtable_' + r_key]
s1 += ['ltable_' + x for x in l_output_attrs if x != l_key]
s1 = sorted(s1)
assert_equal(s1, sorted(C.columns))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(C, 'fk_ltable'), 'ltable_' + l_key)
assert_equal(mg.get_property(C, 'fk_rtable'), 'rtable_' + r_key)
def test_ab_block_candset(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1, r_block_attr_1, l_output_attrs,
r_output_attrs, l_output_prefix, r_output_prefix)
D = self.ab.block_candset(C, l_block_attr_2, r_block_attr_2)
assert_equal(sorted(C.columns), sorted(D.columns))
assert_equal(mg.get_key(D), '_id')
assert_equal(mg.get_property(D, 'fk_ltable'), mg.get_property(C, 'fk_ltable'))
assert_equal(mg.get_property(D, 'fk_rtable'), mg.get_property(C, 'fk_rtable'))
k1 = pd.np.array(D[l_output_prefix + l_block_attr_2])
k2 = pd.np.array(D[r_output_prefix + r_block_attr_2])
assert_equal(all(k1 == k2), True)
def test_reset_property_valid():
df = mg.read_csv_metadata(path_for_A)
mg.set_property(df, 'key', 'ID1')
assert_equal(mg.get_property(df, 'key'), 'ID1')
mg.set_property(df, 'key', 'ID')
assert_equal(mg.get_property(df, 'key'), 'ID')
mg.del_property(df, 'key')
assert_equal(len(mg.get_all_properties(df)), 0)
def test_ab_block_candset_empty_output(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1,
r_block_attr_1)
D = self.ab.block_candset(C, l_block_attr_3, r_block_attr_3)
assert_equal(len(D), 0)
assert_equal(sorted(D.columns), sorted(C.columns))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(D, 'fk_ltable'),
mg.get_property(C, 'fk_ltable'))
assert_equal(mg.get_property(D, 'fk_rtable'),
mg.get_property(C, 'fk_rtable'))
def test_ab_block_tables(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1, r_block_attr_1, l_output_attrs,
r_output_attrs, l_output_prefix, r_output_prefix)
s1 = ['_id', l_output_prefix + l_key, r_output_prefix + r_key]
s1 += [l_output_prefix + x for x in l_output_attrs if x != l_key]
s1 += [r_output_prefix + x for x in r_output_attrs if x != r_key]
s1 = sorted(s1)
assert_equal(s1, sorted(C.columns))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(C, 'fk_ltable'), l_output_prefix + l_key)
assert_equal(mg.get_property(C, 'fk_rtable'), r_output_prefix + r_key)
k1 = pd.np.array(C[l_output_prefix + l_block_attr_1])
k2 = pd.np.array(C[r_output_prefix + r_block_attr_1])
assert_equal(all(k1 == k2), True)
def test_ab_block_candset(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1,
r_block_attr_1, l_output_attrs,
r_output_attrs, l_output_prefix,
r_output_prefix)
D = self.ab.block_candset(C, l_block_attr_2, r_block_attr_2)
assert_equal(sorted(C.columns), sorted(D.columns))
assert_equal(mg.get_key(D), '_id')
assert_equal(mg.get_property(D, 'fk_ltable'),
mg.get_property(C, 'fk_ltable'))
assert_equal(mg.get_property(D, 'fk_rtable'),
mg.get_property(C, 'fk_rtable'))
k1 = pd.np.array(D[l_output_prefix + l_block_attr_2])
k2 = pd.np.array(D[r_output_prefix + r_block_attr_2])
assert_equal(all(k1 == k2), True)
def test_ab_block_tables(self):
C = self.ab.block_tables(self.A, self.B, l_block_attr_1,
r_block_attr_1, l_output_attrs,
r_output_attrs, l_output_prefix,
r_output_prefix)
s1 = ['_id', l_output_prefix + l_key, r_output_prefix + r_key]
s1 += [l_output_prefix + x for x in l_output_attrs if x != l_key]
s1 += [r_output_prefix + x for x in r_output_attrs if x != r_key]
s1 = sorted(s1)
assert_equal(s1, sorted(C.columns))
assert_equal(mg.get_key(C), '_id')
assert_equal(mg.get_property(C, 'fk_ltable'), l_output_prefix + l_key)
assert_equal(mg.get_property(C, 'fk_rtable'), r_output_prefix + r_key)
k1 = pd.np.array(C[l_output_prefix + l_block_attr_1])
k2 = pd.np.array(C[r_output_prefix + r_block_attr_1])
assert_equal(all(k1 == k2), True)
def test_get_property_invalid_no_df():
mg.get_property(None, 'key')
