import unittest

import pandas as pd

from inmoose.diffexp import DEResults

class Test(unittest.TestCase):

    def test_DEResults(self):

        df = pd.DataFrame()

        with self.assertRaisesRegex(

            ValueError, "log2FoldChange missing from results table"

        ):

            res = DEResults(df)

        df["log2FoldChange"] = [1, 2, 3]

        with self.assertRaisesRegex(ValueError, "lfcSE missing from results table"):

            res = DEResults(df)

        df["lfcSE"] = [0.1, 0.2, 0.3]

        with self.assertRaisesRegex(ValueError, "pvalue missing from results table"):

            res = DEResults(df)

        df["pvalue"] = [0.01, 0.02, 0.03]

        res = DEResults(df)

        _subres = res.loc[:1, :]