diff --git a/src/rtichoke/__init__.py b/src/rtichoke/__init__.py index 5e8b0855..d5df5ce7 100644 --- a/src/rtichoke/__init__.py +++ b/src/rtichoke/__init__.py @@ -63,18 +63,26 @@ __all__ = [ "create_roc_curve", + "create_roc_curve_times", "plot_roc_curve", "create_lift_curve", + "create_lift_curve_times", "plot_lift_curve", "create_precision_recall_curve", + "create_precision_recall_curve_times", "plot_precision_recall_curve", "create_gains_curve", + "create_gains_curve_times", "plot_gains_curve", "create_calibration_curve", + "create_calibration_curve_times", "create_decision_curve", + "create_decision_curve_times", "plot_decision_curve", "prepare_performance_data", + "prepare_binned_classification_data", "prepare_performance_data_times", + "prepare_binned_classification_data_times", "create_performance_table", "create_performance_table_times", "render_performance_table", diff --git a/tests/test_public_exports.py b/tests/test_public_exports.py new file mode 100644 index 00000000..422da87e --- /dev/null +++ b/tests/test_public_exports.py @@ -0,0 +1,37 @@ +import rtichoke + + +EXPECTED_PUBLIC_EXPORTS = { + "create_roc_curve", + "create_roc_curve_times", + "plot_roc_curve", + "create_lift_curve", + "create_lift_curve_times", + "plot_lift_curve", + "create_precision_recall_curve", + "create_precision_recall_curve_times", + "plot_precision_recall_curve", + "create_gains_curve", + "create_gains_curve_times", + "plot_gains_curve", + "create_calibration_curve", + "create_calibration_curve_times", + "create_decision_curve", + "create_decision_curve_times", + "plot_decision_curve", + "prepare_performance_data", + "prepare_binned_classification_data", + "prepare_performance_data_times", + "prepare_binned_classification_data_times", + "create_performance_table", + "create_performance_table_times", + "render_performance_table", + "create_summary_report", +} + + +def test_all_lists_every_top_level_public_api(): + assert set(rtichoke.__all__) == EXPECTED_PUBLIC_EXPORTS + + for name in rtichoke.__all__: + assert hasattr(rtichoke, name)