Update tests to support parameter on _get_capacity()

2014-02-08 14:48:20 -07:00 · 2014-02-08 14:48:20 -07:00 · c4fb09a418
parent 18f4fdaa70
commit c4fb09a418
1 changed files with 1 additions and 1 deletions
--- a/tests/test_segments.py
+++ b/tests/test_segments.py
@ -386,7 +386,7 @@ class TestCommon(TestCase):
 	def test_battery(self):
 		pl = Pl()

-		def _get_capacity():
+		def _get_capacity(pl):
 			return 86

 		with replace_attr(common, '_get_capacity', _get_capacity):