use assert instead of assertFalse,assertEqual

Author: nchmura4

pandas/tests/io/formats/test_style.py

@@ -817,19 +817,19 @@ def test_hide_single_index(self):
         assert(ctx['body'][0][0]['is_visible'])
         assert(ctx['head'][0][0]['is_visible'])
         ctx2 = self.df.style.hide_index()._translate()
-        self.assertFalse(ctx2['body'][0][0]['is_visible'])
-        self.assertFalse(ctx2['head'][0][0]['is_visible'])
+        assert(not ctx2['body'][0][0]['is_visible'])
+        assert(not ctx2['head'][0][0]['is_visible'])
 
         # single named index
         ctx3 = self.df.set_index('A').style._translate()
         assert(ctx3['body'][0][0]['is_visible'])
-        self.assertEqual(len(ctx3['head']), 2)  # 2 header levels
+        assert(len(ctx3['head']) == 2)  # 2 header levels
         assert(ctx3['head'][0][0]['is_visible'])
 
         ctx4 = self.df.set_index('A').style.hide_index()._translate()
-        self.assertFalse(ctx4['body'][0][0]['is_visible'])
-        self.assertEqual(len(ctx4['head']), 1)  # only 1 header levels
-        self.assertFalse(ctx4['head'][0][0]['is_visible'])
+        assert(not ctx4['body'][0][0]['is_visible'])
+        assert(len(ctx4['head']) == 1)  # only 1 header levels
+        assert(not ctx4['head'][0][0]['is_visible'])
 
     def test_hide_multiindex(self):
         df = pd.DataFrame({'A': [1, 2]}, index=pd.MultiIndex.from_arrays(
@@ -846,33 +846,33 @@ def test_hide_multiindex(self):
 
         ctx2 = df.style.hide_index()._translate()
         # tests for 'a' and '0'
-        self.assertFalse(ctx2['body'][0][0]['is_visible'])
-        self.assertFalse(ctx2['body'][0][1]['is_visible'])
+        assert(not ctx2['body'][0][0]['is_visible'])
+        assert(not ctx2['body'][0][1]['is_visible'])
         # check for blank header rows
-        self.assertFalse(ctx2['head'][0][0]['is_visible'])
-        self.assertFalse(ctx2['head'][0][1]['is_visible'])
+        assert(not ctx2['head'][0][0]['is_visible'])
+        assert(not ctx2['head'][0][1]['is_visible'])
 
     def test_hide_columns_single_level(self):
         # test hiding single column
         ctx = self.df.style._translate()
         assert(ctx['head'][0][1]['is_visible'])
-        self.assertEqual(ctx['head'][0][1]['display_value'], 'A')
+        assert(ctx['head'][0][1]['display_value'] == 'A')
         assert(ctx['head'][0][2]['is_visible'])
-        self.assertEqual(ctx['head'][0][2]['display_value'], 'B')
+        assert(ctx['head'][0][2]['display_value'] == 'B')
         assert(ctx['body'][0][1]['is_visible'])  # col A, row 1
         assert(ctx['body'][1][2]['is_visible'])  # col B, row 1
 
         ctx = self.df.style.hide_columns('A')._translate()
-        self.assertFalse(ctx['head'][0][1]['is_visible'])
-        self.assertFalse(ctx['body'][0][1]['is_visible'])  # col A, row 1
+        assert(not ctx['head'][0][1]['is_visible'])
+        assert(not ctx['body'][0][1]['is_visible'])  # col A, row 1
         assert(ctx['body'][1][2]['is_visible'])  # col B, row 1
 
         # test hiding mulitiple columns
         ctx = self.df.style.hide_columns(['A', 'B'])._translate()
-        self.assertFalse(ctx['head'][0][1]['is_visible'])
-        self.assertFalse(ctx['head'][0][2]['is_visible'])
-        self.assertFalse(ctx['body'][0][1]['is_visible'])  # col A, row 1
-        self.assertFalse(ctx['body'][1][2]['is_visible'])  # col B, row 1
+        assert(not ctx['head'][0][1]['is_visible'])
+        assert(not ctx['head'][0][2]['is_visible'])
+        assert(not ctx['body'][0][1]['is_visible'])  # col A, row 1
+        assert(not ctx['body'][1][2]['is_visible'])  # col B, row 1
 
     def test_hide_columns_mult_levels(self):
         # setup dataframe with multiple column levels and indices
@@ -887,39 +887,39 @@ def test_hide_columns_mult_levels(self):
         # column headers
         assert(ctx['head'][0][2]['is_visible'])
         assert(ctx['head'][1][2]['is_visible'])
-        self.assertEqual(ctx['head'][1][3]['display_value'], 1)
+        assert(ctx['head'][1][3]['display_value'] == 1)
         # indices
         assert(ctx['body'][0][0]['is_visible'])
         # data
         assert(ctx['body'][1][2]['is_visible'])
-        self.assertEqual(ctx['body'][1][2]['display_value'], 3)
+        assert(ctx['body'][1][2]['display_value'] == 3)
         assert(ctx['body'][1][3]['is_visible'])
-        self.assertEqual(ctx['body'][1][3]['display_value'], 4)
+        assert(ctx['body'][1][3]['display_value'] == 4)
 
         # hide top column level, which hides both columns
         ctx = df.style.hide_columns('b')._translate()
-        self.assertFalse(ctx['head'][0][2]['is_visible'])  # b
-        self.assertFalse(ctx['head'][1][2]['is_visible'])  # 0
-        self.assertFalse(ctx['body'][1][2]['is_visible'])  # 3
+        assert(not ctx['head'][0][2]['is_visible'])  # b
+        assert(not ctx['head'][1][2]['is_visible'])  # 0
+        assert(not ctx['body'][1][2]['is_visible'])  # 3
         assert(ctx['body'][0][0]['is_visible'])  # index
 
         # hide first column only
         ctx = df.style.hide_columns([('b', 0)])._translate()
         assert(ctx['head'][0][2]['is_visible'])  # b
-        self.assertFalse(ctx['head'][1][2]['is_visible'])  # 0
-        self.assertFalse(ctx['body'][1][2]['is_visible'])  # 3
+        assert(not ctx['head'][1][2]['is_visible'])  # 0
+        assert(not ctx['body'][1][2]['is_visible'])  # 3
         assert(ctx['body'][1][3]['is_visible'])
-        self.assertEqual(ctx['body'][1][3]['display_value'], 4)
+        assert(ctx['body'][1][3]['display_value'] == 4)
 
         # hide second column and index
         ctx = df.style.hide_columns([('b', 1)]).hide_index()._translate()
-        self.assertFalse(ctx['body'][0][0]['is_visible'])  # index
+        assert(not ctx['body'][0][0]['is_visible'])  # index
         assert(ctx['head'][0][2]['is_visible'])  # b
         assert(ctx['head'][1][2]['is_visible'])  # 0
-        self.assertFalse(ctx['head'][1][3]['is_visible'])  # 1
-        self.assertFalse(ctx['body'][1][3]['is_visible'])  # 4
+        assert(not ctx['head'][1][3]['is_visible'])  # 1
+        assert(not ctx['body'][1][3]['is_visible'])  # 4
         assert(ctx['body'][1][2]['is_visible'])
-        self.assertEqual(ctx['body'][1][2]['display_value'], 3)
+        assert(ctx['body'][1][2]['display_value'] == 3)
 
 
 class TestStylerMatplotlibDep(object):