Files
python/exercises/practice/pythagorean-triplet/pythagorean_triplet_test.py
BethanyG e7a6b0dc7d [JinJa2] Corrected the macro used for comments on the test file. (#3373)
* Corrected the macro for comments on the test file.
* Added current_date (utcnow()) variable available for template macros.
* Removed unnecessary datetime import from macros file.
* Regenerated all practice exercise test files to add timestamp.
* Changed `datetime.now(tz=timezone.utc)` to `datetime.now(tz=timezone.utc).date()`
* Second regeneration to remove `timestamp` and just keep `date` for test files.
[no important files changed]
2023-07-14 15:52:15 -07:00

54 lines
1.6 KiB
Python

import unittest
from pythagorean_triplet import (
triplets_with_sum,
)
# These tests are auto-generated with test data from:
# https://github.com/exercism/problem-specifications/tree/main/exercises/pythagorean-triplet/canonical-data.json
# File last updated on 2023-07-14
class PythagoreanTripletTest(unittest.TestCase):
def test_triplets_whose_sum_is_12(self):
self.assertCountEqual(triplets_with_sum(12), [[3, 4, 5]])
def test_triplets_whose_sum_is_108(self):
self.assertCountEqual(triplets_with_sum(108), [[27, 36, 45]])
def test_triplets_whose_sum_is_1000(self):
self.assertCountEqual(triplets_with_sum(1000), [[200, 375, 425]])
def test_no_matching_triplets_for_1001(self):
self.assertCountEqual(triplets_with_sum(1001), [])
def test_returns_all_matching_triplets(self):
self.assertCountEqual(triplets_with_sum(90), [[9, 40, 41], [15, 36, 39]])
def test_several_matching_triplets(self):
self.assertCountEqual(
triplets_with_sum(840),
[
[40, 399, 401],
[56, 390, 394],
[105, 360, 375],
[120, 350, 370],
[140, 336, 364],
[168, 315, 357],
[210, 280, 350],
[240, 252, 348],
],
)
def test_triplets_for_large_number(self):
self.assertCountEqual(
triplets_with_sum(30000),
[
[1200, 14375, 14425],
[1875, 14000, 14125],
[5000, 12000, 13000],
[6000, 11250, 12750],
[7500, 10000, 12500],
],
)