#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Testing suite for ARG class.

"""
from __future__ import print_function, division

import unittest as ut
import numpy as np

from skewstudent import SkewStudent

__author__ = "Stanislav Khrapov"
__email__ = "khrapovs@gmail.com"


class ARGTestCase(ut.TestCase):

    """Test SkewStudent distribution class."""

    def test_init(self):
        """Test __init__."""

        skewt = SkewStudent()

        self.assertIsInstance(skewt.nup, float)
        self.assertIsInstance(skewt.lam, float)

        nup, lam = 5., -.2
        skewt = SkewStudent(nup=nup, lam=lam)

        self.assertEqual(skewt.nup, nup)
        self.assertEqual(skewt.lam, lam)

    def test_pdf(self):
        """Test pdf method."""

        skewt = SkewStudent()

        num = 50
        arg = np.linspace(-1, 1, 50)
        pdf = skewt.pdf(arg)

        self.assertEqual(pdf.shape[0], num)
        self.assertIsInstance(skewt.pdf(0), float)


if __name__ == '__main__':
    ut.main()

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Testing suite for SkewStudent class.

"""
from __future__ import print_function, division

import unittest as ut
import numpy as np

from skewstudent import SkewStudent

__author__ = "Stanislav Khrapov"
__email__ = "khrapovs@gmail.com"


class ARGTestCase(ut.TestCase):

    """Test SkewStudent distribution class."""

    def test_init(self):
        """Test __init__."""

        skewt = SkewStudent()

        self.assertIsInstance(skewt.nup, float)
        self.assertIsInstance(skewt.lam, float)

        nup, lam = 5., -.2
        skewt = SkewStudent(nup=nup, lam=lam)

        self.assertEqual(skewt.nup, nup)
        self.assertEqual(skewt.lam, lam)

    def test_pdf(self):
        """Test pdf method."""

        skewt = SkewStudent()

        num = 50
        arg = np.linspace(-1, 1, 50)
        pdf = skewt.pdf(arg)

        self.assertEqual(pdf.shape[0], num)
        self.assertIsInstance(skewt.pdf(0), float)


if __name__ == '__main__':
    ut.main()

Fix title in the test