mirror of
https://github.com/priyanshujain/messageboardbench.git
synced 2026-10-02 19:17:07 +00:00
27 lines
941 B
Python
27 lines
941 B
Python
from func import count_valid_colorings
|
|
|
|
def test_func(candidate):
|
|
# Sample test cases
|
|
assert candidate(3, 3) == 6
|
|
assert candidate(4, 2) == 2
|
|
assert candidate(987654, 456789) == 778634319
|
|
|
|
# Additional test cases
|
|
assert candidate(2, 2) == 2
|
|
assert candidate(1000000, 2) == 2
|
|
assert candidate(458753, 591009) == 0
|
|
assert candidate(2, 1000000) == 756402647
|
|
assert candidate(21180, 536683) == 85696088
|
|
assert candidate(49607, 355757) == 813626735
|
|
assert candidate(303555, 533888) == 997789330
|
|
assert candidate(1000000, 97703) == 868324483
|
|
assert candidate(332578, 1000000) == 940254583
|
|
assert candidate(1000000, 115595) == 893037484
|
|
assert candidate(458753, 591009) == 1603161
|
|
assert candidate(1000000, 580883) == 280492533
|
|
assert candidate(1000000, 1000000) == 201471773
|
|
|
|
if __name__ == "__main__":
|
|
test_func(count_valid_colorings)
|
|
print("All tests passed!")
|