import logging import unittest import numpy as np from src.a_job import id_func logger = logging.getLogger(__name__) class AJobTest(unittest.TestCase): def test_a_job(self): a_vector = np.ndarray([1, 2, 3]) self.assertEqual(id_func(a_vector), a_vector) logger.info("Test passes")