import unittest

import numpy as np

from inmoose.edgepy import DGEList, adjustedProfileLik
from inmoose.utils import rnbinom


class test_APL(unittest.TestCase):
    def setUp(self):
        y = np.array(rnbinom(80, size=5, mu=20, seed=42)).reshape((20, 4))
        y = np.vstack(([0, 0, 0, 0], [0, 0, 2, 2], y))
        self.group = np.array([1, 1, 2, 2])
        self.d = DGEList(counts=y, group=self.group, lib_size=np.arange(1001, 1005))

    def test_adjustedProfileLik(self):
        apl = adjustedProfileLik(
            0.05,
            self.d.counts,
            np.ones((self.d.counts.shape[1], 1)),
            self.d.getOffset(),
        )
        ref = [
            np.inf,
            -6.052019642,
            -14.4426662,
            -12.70072688,
            -23.58414577,
            -17.64428625,
            -15.06042351,
            -18.9616632,
            -12.42791699,
            -18.5664074,
            -15.50008961,
            -13.15651905,
            -18.58032921,
            -16.07588861,
            -14.2002113,
            -15.92987782,
            -12.68396098,
            -24.62550341,
            -17.70222029,
            -15.50752335,
            -14.39845558,
            -13.04470154,
        ]
        self.assertTrue(np.allclose(apl, ref, atol=0, rtol=1e-9))