import unittest
import numpy as np
from inmoose.limma.fitFDist import trigammaInverse
class Test(unittest.TestCase):
def test_trigammainverse(self):
res = trigammaInverse(np.array([np.nan, 0, 1e8, -1, 5]))
self.assertTrue(
np.allclose(
res,
[np.nan, np.inf, 0.0001000, np.nan, 0.4961687],
equal_nan=True,
atol=1e-5,
)
)