test_hello.py

import unittest
import hello


class TestHello(unittest.TestCase):
    def test_hello(self):
        self.assertEqual(hello.hello(), "Hello, world!")
        self.assertNotEqual(hello.hello(), "Hello, friend!")
    
    def test_add(self):
        self.assertEqual(hello.add(1,2), 3)
        self.assertNotEqual(hello.add(1,3),5)
    
    def test_sub(self):
        self.assertEqual(hello.sub(2,1),1)
        self.assertNotEqual(hello.sub(1,3),-3)
    
    def test_mul(self):
        self.assertEqual(hello.mul(3,3), 9)
        self.assertNotEqual(hello.mul(3,4), 9)

    def test_div(self):
        self.assertEqual(hello.div(6,3), 2)

        with self.assertRaises(ValueError):
            hello.div(1,0)

    def test_sqrt(self):
        self.assertEqual(hello.sqrt(9), 3)
        self.assertNotEqual(hello.sqrt(16),2)

    def test_power(self):
        self.assertEqual(hello.power(3,2),9)
        self.assertNotEqual(hello.power(4,3),63)
    
    def test_log(self):
        self.assertEqual(round(hello.log(7.3890561), 1), 2)
        self.assertNotEqual(round(hello.log(54.59815),1),6)

    def test_exp(self):
        self.assertEqual(round(hello.exp(2), 7), 7.3890561)
        self.assertNotEqual(round(hello.exp(5),6),150)

    def test_sin(self):
        self.assertEqual(hello.sin(0), 0)
        self.assertEqual(hello.sin(1), 0.8414709848078965)

    def test_cos(self):
        self.assertEqual(hello.cos(0), 1)
        self.assertEqual(hello.cos(1), 0.5403023058681398)

    def test_tan(self):
        self.assertEqual(hello.tan(0), 0)
        self.assertEqual(hello.tan(1), 1.5574077246549023)

    def test_cot(self):
        self.assertEqual(hello.cot(0), float("inf"))
        self.assertEqual(hello.cot(1), 0.6420926159343306)


if __name__ == "__main__":
    unittest.main()