tests package

Submodules

tests.test_batch_model_predict module

tests.test_batch_model_predict.mock_engine()

Mock the SQLAlchemy engine object

tests.test_batch_model_predict.mock_env_vars(monkeypatch)

Mock environment variables for testing

tests.test_batch_model_predict.mock_model()

Mock the model object

tests.test_batch_model_predict.mock_read_sql()

Mock the pd.read_sql_query function

tests.test_batch_model_predict.mock_to_sql()

Mock the pd.DataFrame.to_sql task

tests.test_batch_model_predict.test_batch_model_predict(mock_model)

Test the batch_model_predict flow

tests.test_batch_model_predict.test_predict(mock_model)

Test the predict task

tests.test_batch_model_predict.test_read_new_data(mock_read_sql, mock_engine)

Test the read_new_data task

tests.test_batch_model_predict.test_save_predictions(mock_to_sql, mock_engine)

Test the save_predictions task

Module contents