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