Seirdy/func-analysis

# -*- coding: utf-8 -*-

"""Test memoization of analyzed functions.

This test is relatively slow and time-consuming, and tests a relatively
unimportant feature (memoization). Run it only if you really need to.
"""
from typing import Tuple

import numpy as np

from func_analysis.tests.call_counting import workout_analyzed_func


def test_call_counting(analyzed_trig_func_counted):
    """Check call counts for each executed function."""
    counts = workout_analyzed_func(analyzed_trig_func_counted)
    original_vals: Tuple[int, ...] = tuple(counts[0].values())
    deduped_vals: Tuple[int, ...] = tuple(counts[1].values())
    repeated_calls = np.subtract(original_vals, deduped_vals)
    counts_after_repeat = counts[0]["dupe"]

    assert original_vals.count(counts_after_repeat) > 1
    assert counts_after_repeat < 1000
    assert not repeated_calls.any()