def test_calculate_max_better(self, seasonal_periods, expected_max_harmonics):
strategy = HarmonicsChoosingStrategy(Context(), checking_range=1)
harmonics = strategy.calculate_max(
np.asarray(seasonal_periods),
HarmonicsChoosingStrategy.max_harmonic_dependency_reduction_better
)
assert np.array_equal(expected_max_harmonics, harmonics)
def test_calculate_max(self, seasonal_periods, expected_max_harmonics):
strategy = HarmonicsChoosingStrategy(Context(), checking_range=1)
harmonics = strategy.calculate_max(np.array(seasonal_periods))
assert np.array_equal(expected_max_harmonics, harmonics)