lomas_server.tests package

Submodules

lomas_server.tests.constants module

lomas_server.tests.test_api module

class lomas_server.tests.test_api.TestRootAPIEndpoint(methodName='runTest')

Bases: TestCase

End-to-end tests of the api endpoints.

This test can be both executed as an integration test (enabled by setting LOMAS_TEST_MONGO_INTEGRATION to True), or a standard test. The first requires a mongodb to be started before running while the latter will use a local YamlDatabase.

setUp() → None

Set Up Header and DB for test.

classmethod setUpClass() → None

Hook method for setting up class fixture before running tests in the class.

tearDown() → None

Hook method for deconstructing the test fixture after testing it.

classmethod tearDownClass() → None

Hook method for deconstructing the class fixture after running all tests in the class.

test_config_and_internal_server_exception() → None

Test set wrong configuration.

test_dummy_opendp_query() → None

Test_dummy_opendp_query.

test_dummy_smartnoise_sql_query() → None

Test_dummy_smartnoise_sql_query.

test_get_dataset_metadata() → None

Test_get_dataset_metadata.

test_get_dummy_dataset() → None

Test_get_dummy_dataset.

test_get_initial_budget() → None

Test_get_initial_budget.

test_get_previous_queries() → None

Test_get_previous_queries.

test_get_remaining_budget() → None

Test_get_remaining_budget.

test_get_total_spent_budget() → None

Test_get_total_spent_budget.

test_opendp_cost() → None

Test_opendp_cost.

test_opendp_query() → None

Test_opendp_query.

test_root() → None

Test root endpoint redirection to state endpoint.

test_smartnoise_sql_cost() → None

Test_smartnoise_sql_cost.

test_smartnoise_sql_query() → None

Test smartnoise-sql query.

test_smartnoise_sql_query_datetime() → None

Test smartnoise-sql query on datetime.

test_smartnoise_sql_query_on_s3_dataset() → None

Test smartnoise-sql on s3 dataset.

test_smartnoise_sql_query_parameters() → None

Test smartnoise-sql query parameters.

test_state() → None

Test state endpoint.

test_subsequent_budget_limit_logic() → None

Test_subsequent_budget_limit_logic.

lomas_server.tests.test_api_diffprivlib module

class lomas_server.tests.test_api_diffprivlib.TestDiffPrivLibEndpoint(methodName='runTest')

Bases: TestRootAPIEndpoint

Test DiffPrivLib Endpoint with different models.

test_clustering_models() → None

Test diffprivlib query: K-Means.

test_diffprivlib_cost() → None

Test_diffprivlib_cost.

test_diffprivlib_query() → None

Test diffprivlib query.

test_dimension_reduction_models() → None

Test diffprivlib query: PCA.

test_dummy_diffprivlib_query() → None

Test_dummy_diffprivlib_query.

test_linear_regression_models() → None

Test diffprivlib query: Linear Regression.

test_logistic_regression_models() → None

Test diffprivlib query: Logistic Regression.

test_naives_bayes_model() → None

Test diffprivlib query: Gaussian Naives Bayes.

test_trees_models() → None

Test diffprivlib query: Random Forest, Decision Tree.

lomas_server.tests.test_api_diffprivlib.validate_pipeline(response) → QueryResponse

Validate that the pipeline ran successfully.

Returns a model and a score.

lomas_server.tests.test_api_smartnoise_synth module

class lomas_server.tests.test_api_smartnoise_synth.TestSmartnoiseSynthEndpoint(methodName='runTest')

Bases: TestRootAPIEndpoint

Test Smartnoise Synth Endpoints with different Synthesizers.

test_dummy_smartnoise_synth_query() → None

Test_dummy_smartnoise_synth_query.

test_smartnoise_synth_cost() → None

Test_smartnoise_synth_cost.

test_smartnoise_synth_query() → None

Test smartnoise synth query.

test_smartnoise_synth_query_aim() → None

Test smartnoise synth query AIM Synthesizer.

test_smartnoise_synth_query_constraints() → None

Test smartnoise synth query constraints.

test_smartnoise_synth_query_datetime() → None

Test smartnoise synth query on other dataset for datetime columns.

test_smartnoise_synth_query_delta_none() → None

Test smartnoise synth query on other synthesizer with delta None.

test_smartnoise_synth_query_dpgan() → None

Test smartnoise synth query dpgan Synthesizer.

test_smartnoise_synth_query_mst() → None

Test smartnoise synth query MST Synthesizer.

test_smartnoise_synth_query_mwem() → None

Test smartnoise synth query MWEM Synthesizer.

test_smartnoise_synth_query_pacsynth() → None

Test smartnoise synth query PAC-Synth Synthesizer.

TOO UNSTABLE BECAUSE OF RUST PANIC

test_smartnoise_synth_query_patectgan() → None

Test smartnoise synth query PATE-CTGAN Synthesizer.

test_smartnoise_synth_query_pategan() → None

Test smartnoise synth query pategan Synthesizer.

test_smartnoise_synth_query_private_id() → None

Test smartnoise synth query on other dataset for private id.

and categorical int columns

test_smartnoise_synth_query_samples() → None

Test smartnoise synth query return samples.

test_smartnoise_synth_query_select_cols() → None

Test smartnoise synth query select_cols.

lomas_server.tests.test_api_smartnoise_synth.validate_response(response) → QueryResponse

Validate that the pipeline ran successfully.

Returns a model and a score.

lomas_server.tests.test_collection_models module

class lomas_server.tests.test_collection_models.TestMetadataModel(methodName='runTest')

Bases: TestCase

Tests for the input validation of dataset metadata.

test_boolean_column() → None

Test_boolean_column.

test_categories() → None

Test categories validation.

test_categories_match_cardinality() → None

Test categories match cardinality.

test_categories_match_type() → None

Test categories match column type.

test_categories_required() → None

Test categories required.

test_datetime_column() → None

Test_datetime_column.

test_float_column() → None

Test_float_column.

test_int_column() → None

Test_int_column.

test_lower_upper_bounded() → None

Test lower is smaller than upper and of right type.

test_metadata_columns_discriminator() → None

Test metadata column discriminator.

test_precision() → None

Test precision can only be 32 or 64.

test_standard_metadata_fields()

Test standard metadata fields.

lomas_server.tests.test_dummy_generation module

class lomas_server.tests.test_dummy_generation.TestMakeDummyDataset(methodName='runTest')

Bases: TestCase

Tests for the generation of dummy datasets.

metadata: dict[str, Any] = {'columns': {}, 'max_ids': 1, 'row_privacy': True, 'rows': 100}

test_boolean_column() → None

Test_boolean_column.

test_categorical_column() → None

Test_categorical_column.

test_datetime_column() → None

Test_datetime_column.

test_float_column() → None

Test_float_column.

test_int_column() → None

Test_int_column.

test_nullable_column() → None

Test_nullable_column.

test_seed() → None

Test_seed.

lomas_server.tests.test_mongodb_admin module

class lomas_server.tests.test_mongodb_admin.TestMongoDBAdmin(methodName='runTest')

Bases: TestCase

Tests for the functions in mongodb_admin.py.

This is an integration test and requires a mongodb database to be started before being executed.

The test is only executed if the LOMAS_TEST_MONGO_INTEGRATION environment variable is set to True.

classmethod setUpClass() → None

Connection to database.

tearDown() → None

Drop all data from database.

test_add_dataset_to_user() → None

Test add dataset to a user.

test_add_datasets_via_yaml() → None

Test add datasets via a YAML file.

test_add_demo_data_to_mongodb_admin() → None

Test add demo data to admin db.

test_add_local_dataset() → None

Test adding a local dataset.

test_add_s3_dataset() → None

Test adding a dataset stored on S3.

test_add_s3_datasets_via_yaml() → None

Test add datasets via a YAML file.

test_add_user() → None

Test adding a user.

test_add_user_wb() → None

Test adding a user with a dataset.

test_add_users_via_yaml() → None

Test create user collection via YAML file.

test_del_dataset() → None

Test dataset deletion.

test_del_dataset_to_user() → None

Test delete dataset from user.

test_del_user() → None

Test deleting a user.

test_drop_collection() → None

Test drop collection from db.

test_get_archives_of_user() → None

Test show archives of user.

test_get_collection() → None

Test show collection from db.

test_get_dataset() → None

Test show dataset.

test_get_list_of_datasets() → None

Test get list of datasets.

test_get_list_of_datasets_from_users() → None

Test get list of datasets from users.

test_get_list_of_users() → None

Test get list of users.

test_get_metadata_of_dataset() → None

Test show metadata_dataset.

test_get_user() → None

Test show user.

test_set_budget_field() → None

Test setting a budget field.

test_set_may_query() → None

Test set may query.

lomas_server.tests.test_mongodb_admin_cli module

class lomas_server.tests.test_mongodb_admin_cli.TestMongoDBAdmin(methodName='runTest')

Bases: TestCase

Tests for the functions in mongodb_admin.py.

This is an integration test and requires a mongodb database to be started before being executed.

The test is only executed if the LOMAS_TEST_MONGO_INTEGRATION environment variable is set to True.

run_cli_command(command: str, args: List) → None

Run a MongoDB administration CLI command.

Parameters:

• command (str) – The subcommand to run.

• args (List[str]) – A list of arguments for the subcommand.

Raises:

ValueError – If the command returns a non-zero exit status.

classmethod setUpClass() → None

Connection to database.

tearDown() → None

Drop all data from database.

test_add_dataset_to_user_cli() → None

Test add dataset to a user via cli.

test_add_datasets_via_yaml_cli() → None

Test add datasets via a YAML file via cli.

test_add_local_dataset_cli() → None

Test adding a local dataset via cli.

test_add_user_cli() → None

Test adding a user via cli.

test_add_user_wb_cli() → None

Test adding a user with a dataset via cli.

test_add_users_via_yaml_cli() → None

Test create user collection via YAML file via cli.

test_del_dataset_cli() → None

Test dataset deletion via cli.

test_del_dataset_to_user_cli() → None

Test delete dataset from user via cli.

test_del_user_cli() → None

Test deleting a user via cli.

test_drop_collection_cli() → None

Test drop collection from db via cli.

test_get_archives_of_user_cli() → None

Test show archives of user via CLI.

Does not verify output for not

test_get_collection_cli() → None

Test show collection from db via CLI.

test_get_dataset_cli() → None

Test show dataset.

Does not verify output for not

test_get_list_of_datasets_cli() → None

Test get list of datasets via CLI.

Does not verify output for not

test_get_list_of_datasets_from_user_cli() → None

Test get list of users via CLI.

Does not verify output for not

test_get_list_of_users_cli() → None

Test get list of users via CLI.

Does not verify output for not

test_get_metadata_of_dataset_cli() → None

Test show metadata_of dataset.

Does not verify output for not

test_get_user_cli() → None

Test show user via CLI.

Does not verify output for not

test_set_budget_field_cli() → None

Test setting a budget field via cli.

test_set_may_query_cli() → None

Test set may query via cli.

Module contents