63180d7e24
* pre-commit autoupdate 2025-09-11 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
import unittest
|
|
|
|
import pytest
|
|
|
|
from knapsack import greedy_knapsack as kp
|
|
|
|
|
|
class TestClass(unittest.TestCase):
|
|
"""
|
|
Test cases for knapsack
|
|
"""
|
|
|
|
def test_sorted(self):
|
|
"""
|
|
kp.calc_profit takes the required argument (profit, weight, max_weight)
|
|
and returns whether the answer matches to the expected ones
|
|
"""
|
|
profit = [10, 20, 30, 40, 50, 60]
|
|
weight = [2, 4, 6, 8, 10, 12]
|
|
max_weight = 100
|
|
assert kp.calc_profit(profit, weight, max_weight) == 210
|
|
|
|
def test_negative_max_weight(self):
|
|
"""
|
|
Returns ValueError for any negative max_weight value
|
|
:return: ValueError
|
|
"""
|
|
# profit = [10, 20, 30, 40, 50, 60]
|
|
# weight = [2, 4, 6, 8, 10, 12]
|
|
# max_weight = -15
|
|
pytest.raises(ValueError, match=r"max_weight must greater than zero.")
|
|
|
|
def test_negative_profit_value(self):
|
|
"""
|
|
Returns ValueError for any negative profit value in the list
|
|
:return: ValueError
|
|
"""
|
|
# profit = [10, -20, 30, 40, 50, 60]
|
|
# weight = [2, 4, 6, 8, 10, 12]
|
|
# max_weight = 15
|
|
pytest.raises(ValueError, match=r"Weight can not be negative.")
|
|
|
|
def test_negative_weight_value(self):
|
|
"""
|
|
Returns ValueError for any negative weight value in the list
|
|
:return: ValueError
|
|
"""
|
|
# profit = [10, 20, 30, 40, 50, 60]
|
|
# weight = [2, -4, 6, -8, 10, 12]
|
|
# max_weight = 15
|
|
pytest.raises(ValueError, match=r"Profit can not be negative.")
|
|
|
|
def test_null_max_weight(self):
|
|
"""
|
|
Returns ValueError for any zero max_weight value
|
|
:return: ValueError
|
|
"""
|
|
# profit = [10, 20, 30, 40, 50, 60]
|
|
# weight = [2, 4, 6, 8, 10, 12]
|
|
# max_weight = null
|
|
pytest.raises(ValueError, match=r"max_weight must greater than zero.")
|
|
|
|
def test_unequal_list_length(self):
|
|
"""
|
|
Returns IndexError if length of lists (profit and weight) are unequal.
|
|
:return: IndexError
|
|
"""
|
|
# profit = [10, 20, 30, 40, 50]
|
|
# weight = [2, 4, 6, 8, 10, 12]
|
|
# max_weight = 100
|
|
pytest.raises(
|
|
IndexError, match=r"The length of profit and weight must be same."
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|