mirror of
https://github.com/donnemartin/interactive-coding-challenges
synced 2026-01-07 18:08:01 +00:00
@@ -1,12 +1,12 @@
|
||||
from nose.tools import assert_equal, assert_raises
|
||||
import unittest
|
||||
|
||||
|
||||
class TestKnapsack(object):
|
||||
class TestKnapsack(unittest.TestCase):
|
||||
|
||||
def test_knapsack_bottom_up(self):
|
||||
knapsack = Knapsack()
|
||||
assert_raises(TypeError, knapsack.fill_knapsack, None, None)
|
||||
assert_equal(knapsack.fill_knapsack(0, 0), 0)
|
||||
self.assertRaises(TypeError, knapsack.fill_knapsack, None, None)
|
||||
self.assertEqual(knapsack.fill_knapsack(0, 0), 0)
|
||||
items = []
|
||||
items.append(Item(label='a', value=2, weight=2))
|
||||
items.append(Item(label='b', value=4, weight=2))
|
||||
@@ -15,18 +15,18 @@ class TestKnapsack(object):
|
||||
total_weight = 8
|
||||
expected_value = 13
|
||||
results = knapsack.fill_knapsack(items, total_weight)
|
||||
assert_equal(results[0].label, 'd')
|
||||
assert_equal(results[1].label, 'b')
|
||||
self.assertEqual(results[0].label, 'd')
|
||||
self.assertEqual(results[1].label, 'b')
|
||||
total_value = 0
|
||||
for item in results:
|
||||
total_value += item.value
|
||||
assert_equal(total_value, expected_value)
|
||||
self.assertEqual(total_value, expected_value)
|
||||
print('Success: test_knapsack_bottom_up')
|
||||
|
||||
def test_knapsack_top_down(self):
|
||||
knapsack = KnapsackTopDown()
|
||||
assert_raises(TypeError, knapsack.fill_knapsack, None, None)
|
||||
assert_equal(knapsack.fill_knapsack(0, 0), 0)
|
||||
self.assertRaises(TypeError, knapsack.fill_knapsack, None, None)
|
||||
self.assertEqual(knapsack.fill_knapsack(0, 0), 0)
|
||||
items = []
|
||||
items.append(Item(label='a', value=2, weight=2))
|
||||
items.append(Item(label='b', value=4, weight=2))
|
||||
@@ -34,7 +34,7 @@ class TestKnapsack(object):
|
||||
items.append(Item(label='d', value=9, weight=5))
|
||||
total_weight = 8
|
||||
expected_value = 13
|
||||
assert_equal(knapsack.fill_knapsack(items, total_weight), expected_value)
|
||||
self.assertEqual(knapsack.fill_knapsack(items, total_weight), expected_value)
|
||||
print('Success: test_knapsack_top_down')
|
||||
|
||||
def main():
|
||||
@@ -44,4 +44,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user