from unittest import TestCase
from benchmark.gcd import gcd


class Test_gcd(TestCase):
    #  distances_true = {1: [6], 2: [3], 3: [3], 4: [4], 5: [0, 0, 0, 1]}
    #  distances_false = {1: [0], 2: [0], 3: [0], 4: [0], 5: [4, 3, 1, 0]}
    def test_gcd_1(self):
        assert gcd(a=7, b=4) == 1

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

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

    #  distances_true = {1: [6], 2: [6], 3: [0]}
    #  distances_false = {1: [0], 2: [0], 3: [1]}
    def test_gcd_4(self):
        assert gcd(a=7, b=7) == 7

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