Fix bug in default_data test exposed by tc 0.15.+

codekansas · Jan 23, 2018 · 4f3e9a6 · 4f3e9a6
1 parent 5959551
commit 4f3e9a6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/webapp/apps/taxbrain/tests/test_all.py b/webapp/apps/taxbrain/tests/test_all.py
@@ -201,12 +201,11 @@ def test_default_taxcalc_data(self):
         floored_std_aged = list(map(math.floor, dd['_STD_Aged'][0]))
         assert dd['_STD_Aged'] == [floored_std_aged]
         assert dd_meta['_STD_Aged']['value'] == [floored_std_aged]
-
         floored_ii_em_ps = list(map(math.floor, dd['_II_em_ps'][0]))
         assert dd['_II_em_ps'] == [floored_ii_em_ps]
         assert dd_meta['_II_em_ps']['value'] == [floored_ii_em_ps]
 
-        floored_ii_em = [math.floor(dd['_II_em'][0])]
+        floored_ii_em = list(map(math.floor, dd['_II_em']))
         assert dd['_II_em'] == floored_ii_em
         assert dd_meta['_II_em']['value'] == floored_ii_em