19 lines
1 KiB
Python
19 lines
1 KiB
Python
from unittest import TestCase
|
|
from benchmark.common_divisor_count import cd_count
|
|
|
|
|
|
class Test_cd_count(TestCase):
|
|
# distances_true = {1: [180], 2: [450], 3: [0], 4: [451], 5: [0, 0, 1], 6: [0, 0, 0, 2, 0, 0, 6, 2, 0], 7: [89, 43, 27, 13, 9, 1]}
|
|
# distances_false = {1: [0], 2: [0], 3: [180], 4: [0], 5: [180, 90, 0], 6: [1, 1, 1, 0, 1, 1, 0, 0, 1], 7: [0, 0, 0, 0, 0, 0]}
|
|
def test_cd_count_1(self):
|
|
assert cd_count(a=-180, b=450) == 12
|
|
|
|
# distances_true = {1: [524], 2: [858], 3: [525], 4: [0], 5: [0, 0, 0, 0, 0, 0, 0, 0, 1], 6: [0], 7: [1]}
|
|
# distances_false = {1: [0], 2: [0], 3: [0], 4: [858], 5: [524, 334, 190, 144, 46, 6, 4, 2, 0], 6: [1], 7: [0]}
|
|
def test_cd_count_2(self):
|
|
assert cd_count(a=524, b=-858) == 2
|
|
|
|
# distances_true = {1: [171], 2: [880], 3: [0], 4: [881], 5: [0, 0, 0, 0, 0, 1], 6: [0], 7: [0]}
|
|
# distances_false = {1: [0], 2: [0], 3: [171], 4: [0], 5: [171, 25, 21, 4, 1, 0], 6: [1], 7: [1]}
|
|
def test_cd_count_3(self):
|
|
assert cd_count(a=-171, b=880) == 1
|