alibi.tests.utils module

class alibi.tests.utils.MockPredictor(out_dim, out_type='proba', model_type=None, seed=None)

Bases: object

A class the mimicks the output of a classifier or regressor to allow testing of functionality that depends on it without inference overhead.

__init__(out_dim, out_type='proba', model_type=None, seed=None)

Parameters:

• out_dim (int) – The number of output classes.

• out_type (str) – Indicates if probabilities, class predictions or continuous outputs are generated.

predict(*args, **kwargs)

alibi.tests.utils.assert_message_in_logs(msg, records)

Helper function to check if a msg is present in any of the records (an iterable of strings).

alibi.tests.utils.issorted(arr, reverse=False)

Checks if a numpy array is sorted.

alibi.tests.utils.not_raises(ExpectedException)

A context manager used to check that ExpectedException does not occur during testing.