def test_ring():
run(IdentitySynthesizer, iterations=1, datasets=['ring'])
def test_mnist28():
run(IdentitySynthesizer, iterations=1, datasets=['mnist28'])
def test_news():
run(IdentitySynthesizer, iterations=1, datasets=['news'])
def test_insurance():
run(IdentitySynthesizer, iterations=1, datasets=['insurance'])
def test_intrusion():
run(IdentitySynthesizer, iterations=1, datasets=['intrusion'])
def test_credit():
run(IdentitySynthesizer, iterations=1, datasets=['credit'])
def test_gridr():
run(IdentitySynthesizer, iterations=1, datasets=['gridr'])
def test_child():
run(IdentitySynthesizer, iterations=1, datasets=['child'])
def test_covtype():
run(IdentitySynthesizer, iterations=1, datasets=['covtype'])
def test_census():
run(IdentitySynthesizer, iterations=1, datasets=['census'])
def test_asia():
run(IdentitySynthesizer, iterations=1, datasets=['asia'])
def test_alarm():
run(IdentitySynthesizer, iterations=1, datasets=['alarm'])
def test_adult():
run(IdentitySynthesizer, iterations=1, datasets=['adult'])
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)
run(synthesizer.fit_sample, datasets=['adult'], repeat=1)
for synthesizer_class in NO_INIT:
synthesizer = synthesizer_class()
run(synthesizer.fit_sample, datasets=['adult'], repeat=1)
logging.info('All the synthesizers were executed successfully')