1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
# holidays
# --------
# A fast, efficient Python library for generating country, province and state
# specific sets of holidays on the fly. It aims to make determining whether a
# specific date is a holiday as fast and flexible as possible.
#
# Authors: Vacanza Team and individual contributors (see CONTRIBUTORS file)
# dr-prodigy <dr.prodigy.github@gmail.com> (c) 2017-2023
# ryanss <ryanssdev@icloud.com> (c) 2014-2017
# Website: https://github.com/vacanza/holidays
# License: MIT (see LICENSE file)
from unittest import TestCase
import numpy as np
from holidays.countries.cambodia import Cambodia
from holidays.countries.thailand import Thailand
from holidays.countries.ukraine import Ukraine
class TestNumpy(TestCase):
def test_years_int_conversion(self):
years = (1995, 2000)
years_range = set(range(*years))
for cls in (Cambodia, Thailand, Ukraine):
# Test single value.
for int_x in (np.int16, np.int32, np.int64):
self.assertEqual(cls(years=int_x(2024)).years, {2024})
# Test iterable.
self.assertEqual(cls(years=np.arange(*years)).years, years_range)
|
