/*
* ARX: Powerful Data Anonymization
* Copyright 2012 - 2017 Fabian Prasser, Florian Kohlmayer and contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.deidentifier.arx.test;
import org.deidentifier.arx.AttributeType.Hierarchy;
import org.deidentifier.arx.AttributeType.Hierarchy.DefaultHierarchy;
import org.deidentifier.arx.Data;
import org.deidentifier.arx.Data.DefaultData;
/**
* Provides data for test cases.
*
* @author Fabian Prasser
* @author Florian Kohlmayer
*/
public class DataProvider {
/** Data */
protected DefaultData data;
/** Hierarchy */
protected DefaultHierarchy age;
/** Hierarchy */
private final DefaultHierarchy ageWrong;
/** Hierarchy */
private final DefaultHierarchy gender;
/** Hierarchy */
private final DefaultHierarchy zipcode;
/** Hierarchy */
private final DefaultHierarchy ageOne;
/** Hierarchy */
private final DefaultHierarchy genderOne;
/** Hierarchy */
private final DefaultHierarchy zipcodeOne;
/** Hierarchy */
private final DefaultHierarchy ageMissing;
/**
* Init.
*/
public DataProvider() {
// Define data
data = Data.create();
data.add("age", "gender", "zipcode");
data.add("34", "male", "81667");
data.add("45", "female", "81675");
data.add("66", "male", "81925");
data.add("70", "female", "81931");
data.add("34", "female", "81931");
data.add("70", "male", "81931");
data.add("45", "male", "81931");
// Define hierarchies
age = Hierarchy.create();
age.add("34", "<50", "*");
age.add("45", "<50", "*");
age.add("66", ">=50", "*");
age.add("70", ">=50", "*");
// Define hierarchies
ageWrong = Hierarchy.create();
ageWrong.add("34", "30-40", "30-69", "*");
ageWrong.add("45", "40-50", "30-69", "*");
ageWrong.add("66", "70-80", "30-69", "*");
ageWrong.add("70", "70-80", "70+", "*");
gender = Hierarchy.create();
gender.add("male", "*");
gender.add("female", "*");
// Only excerpts for readability
zipcode = Hierarchy.create();
zipcode.add("81667", "8166*", "816**", "81***", "8****", "*****");
zipcode.add("81675", "8167*", "816**", "81***", "8****", "*****");
zipcode.add("81925", "8192*", "819**", "81***", "8****", "*****");
zipcode.add("81931", "8193*", "819**", "81***", "8****", "*****");
// Define hierarchies with height one
ageOne = Hierarchy.create();
ageOne.add("34");
ageOne.add("45");
ageOne.add("66");
ageOne.add("70");
genderOne = Hierarchy.create();
genderOne.add("male");
genderOne.add("female");
zipcodeOne = Hierarchy.create();
zipcodeOne.add("81667");
zipcodeOne.add("81675");
zipcodeOne.add("81925");
zipcodeOne.add("81931");
// Define hierarchies
ageMissing = Hierarchy.create();
ageMissing.add("34", "<50", "*");
ageMissing.add("45", "<50", "*");
ageMissing.add("70", ">=50", "*");
}
/**
* Returns a standard data definition.
*/
public void createDataDefinition() {
// Create a standard definition
data.getDefinition().setAttributeType("age", age);
data.getDefinition().setAttributeType("gender", gender);
data.getDefinition().setAttributeType("zipcode", zipcode);
}
/**
* Returns a standard data definition.
*/
public void createDataDefinitionMissing() {
// Create a standard definition
data.getDefinition().setAttributeType("age", ageMissing);
data.getDefinition().setAttributeType("gender", gender);
data.getDefinition().setAttributeType("zipcode", zipcode);
}
/**
* Returns a standard data definition.
*/
public void createDataDefinitionWithHeightOne() {
// Create a standard definition
data.getDefinition().setAttributeType("age", ageOne);
data.getDefinition().setAttributeType("gender", genderOne);
data.getDefinition().setAttributeType("zipcode", zipcodeOne);
}
/**
* Returns a standard data definition.
*/
public void createWrongDataDefinition() {
// Create a standard definition
data.getDefinition().setAttributeType("age", ageWrong);
data.getDefinition().setAttributeType("gender", gender);
data.getDefinition().setAttributeType("zipcode", zipcode);
}
/**
* @return the age
*/
public DefaultHierarchy getAge() {
return age;
}
/**
* @return the data
*/
public DefaultData getData() {
return data;
}
/**
* @return the gender
*/
public DefaultHierarchy getGender() {
return gender;
}
/**
* @return the zipcode
*/
public DefaultHierarchy getZipcode() {
return zipcode;
}
}