TOSQUASH: update DataFrame tests

Author: Martin Journois

pandas/tests/frame/test_api.py

@@ -131,15 +131,19 @@ def test_column_contains_typeerror(self):
     def test_tab_completion(self):
         # DataFrame whose columns are identifiers shall have them in __dir__.
         df = pd.DataFrame([list('abcd'), list('efgh')], columns=list('ABCD'))
-        assert 'A' in dir(df)
+        for key in list('ABCD'):
+            assert key in dir(df)
         assert isinstance(df.__getitem__('A'), pd.Series)
 
         # DataFrame whose first-level columns are identifiers shall have
         # them in __dir__.
         df = pd.DataFrame(
             [list('abcd'), list('efgh')],
             columns=pd.MultiIndex.from_tuples(list(zip('ABCD', 'EFGH'))))
-        assert 'A' in dir(df)
+        for key in list('ABCD'):
+            assert key in dir(df)
+        for key in list('EFGH'):
+            assert key not in dir(df)
         assert isinstance(df.__getitem__('A'), pd.DataFrame)
 
     def test_not_hashable(self):