'''
Created on Sep 7, 2018

@author: haidong
'''
import unittest

from sources.fractional_knapsack import getBestItem, getOptimalValue

class Test(unittest.TestCase):

    def testBestItem1(self):
        values = [60, 100, 120]
        weights = [20, 50, 30]
        self.assertEqual(2, getBestItem(values, weights))

    def testBestItem2(self):
        values = [500]
        weights = [30]
        self.assertEqual(0, getBestItem(values, weights))
    
    def test1(self):
        values = [44, 26,31]
        weights = [6, 28,38]
        capacity = 50
        self.assertAlmostEqual(83.0526, getOptimalValue(capacity, values, weights), 4)
        
    def test2(self):
        values = [60, 100, 120]
        weights = [20,50,30]
        capacity = 50
        self.assertAlmostEqual(180, getOptimalValue(capacity, values, weights), 4)

if __name__ == "__main__":
    # import sys;sys.argv = ['', 'Test.testName']
    unittest.main()