def get_partial_pii_table():
partial_pii_table = Table("test_store", "partial_pii")
partial_pii_a = Column("a")
partial_pii_a.add_pii_type(PiiTypes.PHONE)
partial_pii_b = Column("b")
partial_pii_table.add_child(partial_pii_a)
partial_pii_table.add_child(partial_pii_b)
return partial_pii_table
def get_full_pii_table():
full_pii_table = Table("test_store", "full_pii")
full_pii_a = Column("a")
full_pii_a.add_pii_type(PiiTypes.PHONE)
full_pii_b = Column("b")
full_pii_b.add_pii_type(PiiTypes.ADDRESS)
full_pii_b.add_pii_type(PiiTypes.LOCATION)
full_pii_table.add_child(full_pii_a)
full_pii_table.add_child(full_pii_b)
return full_pii_table