test_a_job.py

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")