Simple unit test 2

unit_test_separated_main.py

import unittest
import sum

class ExampleTestUnit(unittest.TestCase):
  def test_sum(self):
    self.assertEqual(sum.sum_values([1, 2, 3]), 6, "Hasilnya adalah 6")

  def test_error(self):
    with self.assertRaises(TypeError):
      sum.sum_values([1, "2", "3"])

if __name__ == "__main__":
  suite = unittest.TestLoader().loadTestsFromTestCase(ExampleTestUnit)
  unittest.TextTestRunner(verbosity=2).run(suite)