mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-16 18:51:14 +00:00
3f094fe49d
* Python linting: Add ruff rules for Pandas-vet and Pytest-style
* updating DIRECTORY.md
---------
Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
76 lines
2.3 KiB
Python
76 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="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="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="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="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="The length of profit and weight must be same.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|