from unittest import TestCase

from benchmark.gcd import gcd


class Test_gcd(TestCase):
    def test_gcd_1(self):
        assert gcd(a=531, b=338) == 1

    def test_gcd_2(self):
        assert gcd(a=1, b=395) == 1

    def test_gcd_3(self):
        assert gcd(a=71, b=496) == 1

    def test_gcd_4(self):
        assert gcd(a=644, b=644) == 644

    def test_gcd_5(self):
        assert gcd(a=581, b=1) == 1