from unittest import TestCase
from benchmark.gcd import gcd


class Test_gcd(TestCase):
    #  distances_true = {1: [97], 2: [0]}
    #  distances_false = {1: [0], 2: [1]}
    def test_gcd_1(self):
        assert gcd(a=98, b=1) == 1

    #  distances_true = {1: [97], 2: [1], 3: [96], 4: [97], 5: [0, 1]}
    #  distances_false = {1: [0], 2: [0], 3: [0], 4: [0], 5: [2, 0]}
    def test_gcd_2(self):
        assert gcd(a=98, b=2) == 2

    #  distances_true = {1: [9], 2: [9], 3: [0]}
    #  distances_false = {1: [0], 2: [0], 3: [1]}
    def test_gcd_3(self):
        assert gcd(a=10, b=10) == 10

    #  distances_true = {1: [7], 2: [943], 3: [936], 4: [0], 5: [0, 1]}
    #  distances_false = {1: [0], 2: [0], 3: [0], 4: [936], 5: [8, 0]}
    def test_gcd_4(self):
        assert gcd(a=8, b=944) == 8

    #  distances_true = {1: [0]}
    #  distances_false = {1: [1]}
    def test_gcd_5(self):
        assert gcd(a=1, b=933) == 1