view tests/test_sigmoid.py @ 24:89f46435a9e2

[logistic regression] call cost function
author Jeff Hammel <k0scist@gmail.com>
date Mon, 04 Sep 2017 09:58:01 -0700
parents d6d2ecb33c95
children 38aa9098bf2d
line wrap: on
line source

#!/usr/bin/env python

"""
test sigmoid related functionality
"""

import os
import unittest
from tvii.sigmoid import sigmoid


class TestSigmoid(unittest.TestCase):

    def test_basic(self):
        """test two points of the sigmoid function"""

        answer = sigmoid([0,2])
        assert len(answer) == 2
        assert answer[0] == 0.5
        assert abs(answer[1] - 0.88079708) < 1e-6

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