
import unittest
import metric


class MetricTest(unittest.TestCase):


    def test_info_gain(self):
        self.assertEqual(round(metric.calc_gain([9,5]),2), 0.940)
        

if __name__ == "__main__":
    unittest.main()
aee1118f50508bc7882bd8e5aa71bea2cda25890
<mask token>


class MetricTest(unittest.TestCase):
    <mask token>


<mask token>

<mask token>


class MetricTest(unittest.TestCase):

    def test_info_gain(self):
        self.assertEqual(round(metric.calc_gain([9, 5]), 2), 0.94)


<mask token>

<mask token>


class MetricTest(unittest.TestCase):

    def test_info_gain(self):
        self.assertEqual(round(metric.calc_gain([9, 5]), 2), 0.94)


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

import unittest
import metric


class MetricTest(unittest.TestCase):

    def test_info_gain(self):
        self.assertEqual(round(metric.calc_gain([9, 5]), 2), 0.94)


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


import unittest
import metric


class MetricTest(unittest.TestCase):


    def test_info_gain(self):
        self.assertEqual(round(metric.calc_gain([9,5]),2), 0.940)
        

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