import unittest

import numpy as np

import patsy

from inmoose.deseq2 import DESeq, makeExampleDESeqDataSet

from inmoose.utils import Factor

@unittest.skip("lfcShrink is not implemented")

class Test(unittest.TestCase):

    def test_interation_errors(self):

        """test that interactions throw errors"""

        dds = makeExampleDESeqDataSet(n=100, m=8)

        dds.obs["group"] = Factor(np.repeat(["X", "Y"], dds.n_obs / 2))

        dds.design = "~ condition + group + condition:group"

        dds = DESeq(dds)

        self.assertEqual(dds.resultsNames()[3], "conditionB.groupY")

        # interactions error

        with self.assertRaisesRegex(

            ValueError, expected_regex="designs with interations"

        ):

            DESeq(dds, betaPrior=True)

        # also lfcShrink

        res = dds.results(name="conditionB.groupY")

        with self.assertRaises(NotImplementedError):

            res = lfcShrink(dds, coef=4, res=res, type="normal")  # noqa: F821

        res = dds.results(contrast=["condition", "B", "A"])

        with self.assertRaises(NotImplementedError):

            res = lfcShrink(  # noqa: F821

                dds, contrast=["condition", "B", "A"], res=res, type="normal"

            )

        # however, this is allowed

        dds2 = dds.copy()

        dds2.design = patsy.dmatrix("~ condition + group + condition:group", dds2.obs)

        dds2 = DESeq(dds2)

        dds2.results(name="conditionB.groupY")

        lfcShrink(dds2, coef=4, res=res, type="normal")  # noqa: F821