def test_adult():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['adult'])
def test_adult():
benchmark(IdentitySynthesizer, iterations=1, datasets=['adult'])
def test_intrusion():
benchmark(IdentitySynthesizer, iterations=1, datasets=['intrusion'])
def test_news():
benchmark(IdentitySynthesizer, iterations=1, datasets=['news'])
def test_covtype():
benchmark(IdentitySynthesizer, iterations=1, datasets=['covtype'])
def test_gridr():
benchmark(IdentitySynthesizer, iterations=1, datasets=['gridr'])
def test_credit():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['credit'])
def test_census():
benchmark(IdentitySynthesizer, iterations=1, datasets=['census'])
def test_child():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['child'])
def test_covtype():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['covtype'])
def test_census():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['census'])
def test_asia():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['asia'])
def test_alarm():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['alarm'])
def test_alarm():
benchmark(IdentitySynthesizer, iterations=1, datasets=['alarm'])
def test_gridr():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['gridr'])
def test_asia():
benchmark(IdentitySynthesizer, iterations=1, datasets=['asia'])
def test_insurance():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['insurance'])
def test_child():
benchmark(IdentitySynthesizer, iterations=1, datasets=['child'])
def test_intrusion():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['intrusion'])
def test_credit():
benchmark(IdentitySynthesizer, iterations=1, datasets=['credit'])
def test_mnist28():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['mnist28'])
def test_insurance():
benchmark(IdentitySynthesizer, iterations=1, datasets=['insurance'])
def test_news():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['news'])
def test_mnist28():
benchmark(IdentitySynthesizer, iterations=1, datasets=['mnist28'])
VEEGANSynthesizer,
)
NO_INIT = (
CLBNSynthesizer,
IndependentSynthesizer,
IdentitySynthesizer,
UniformSynthesizer,
PrivBNSynthesizer,
)
if __name__ == '__main__':
# This is to be run locally by hand, as some synthesizers take
# a long time and might fail in travis
#
# Run as:
#
# $ python tests/test_synthesizers.py
logging.basicConfig(level=logging.INFO)
for synthesizer_class in EPOCHS_SYNTHS:
synthesizer = synthesizer_class(epochs=1)
benchmark(synthesizer.fit_sample, datasets=['adult'], repeat=1)
for synthesizer_class in NO_INIT:
synthesizer = synthesizer_class()
benchmark(synthesizer.fit_sample, datasets=['adult'], repeat=1)
logging.info('All the synthesizers were executed successfully')
def test_ring():
benchmark(IdentitySynthesizer, iterations=1, datasets=['ring'])
def test_ring():
benchmark(synthesizer.fit_sample, repeat=1, datasets=['ring'])
