Exemplos de mem_profile em Python

Exemplo n.º 1

Arquivo: test_memory_profiler.py Projeto: zeta1999/kedro

def test_mem_profile(caplog):
    func = memory_profiler.mem_profile(sleeping_identity)
    res = func(1)

    logger_name, severity, message = caplog.record_tuples[0]
    assert res == 1
    assert logger_name == "kedro.extras.decorators.memory_profiler"
    assert severity == logging.INFO
    expected = "Running '{}.{}' consumed".format(
        sleeping_identity.__module__, sleeping_identity.__qualname__)
    assert expected in message

Exemplo n.º 2

Arquivo: test_memory_profiler.py Projeto: zeta1999/kedro

def test_mem_profile_old_versions(caplog, mocker):
    mocker.patch(
        "kedro.extras.decorators.memory_profiler.memory_usage",
        return_value=[[float(0)], 1],
    )
    func = memory_profiler.mem_profile(sleeping_identity)
    res = func(1)

    logger_name, severity, message = caplog.record_tuples[0]
    assert res == 1
    assert logger_name == "kedro.extras.decorators.memory_profiler"
    assert severity == logging.INFO
    expected = "Running '{}.{}' consumed".format(
        sleeping_identity.__module__, sleeping_identity.__qualname__)
    assert expected in message