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_smartnoise_sql_query() → None

test_get_dataset_metadata() → None

test_get_dummy_dataset() → None

test_get_initial_budget() → None

test_get_previous_queries() → None

test_get_remaining_budget() → None

test_get_total_spent_budget() → None

test_opendp_cost() → None

test_opendp_query() → None

test_root() → None

Test root endpoint redirection to state endpoint

test_smartnoise_sql_cost() → None

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

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_query() → None

Test diffprivlib query

test_dimension_reduction_models() → None

Test diffprivlib query: PCA

test_dummy_diffprivlib_query() → None

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)

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_smartnoise_synth_cost() → None

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.get_model(query_response)

Unpickle model from API response

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_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_float_column() → None

test_int_column() → None

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_categorical_column() → None

test_datetime_column() → None

test_float_column() → None

test_int_column() → None

test_nullable_column() → None

test_seed() → None

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