fixed tests

Chilipp · May 15, 2018 · b7e6aae · b7e6aae
1 parent 9569c95
commit b7e6aae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/widgets/test_menu_actions.py b/tests/widgets/test_menu_actions.py
@@ -97,7 +97,7 @@ def test_export_final(self):
         fname = self.get_random_filename(suffix='.csv')
         self.straditizer_widgets.menu_actions.export_final(fname)
         self.assertTrue(osp.exists(fname), msg=fname + ' is missing!')
-        exported = pd.read_csv(fname, index_col=0)
+        exported = pd.read_csv(fname, index_col=0, comment='#')
         exported.columns = self.reader._full_df.columns
         self.assertFrameEqual(exported, self.straditizer.final_df)
 
@@ -109,7 +109,7 @@ def test_export_full_df(self):
         fname = self.get_random_filename(suffix='.csv')
         self.straditizer_widgets.menu_actions.export_full(fname)
         self.assertTrue(osp.exists(fname), msg=fname + ' is missing!')
-        exported = pd.read_csv(fname, index_col=0)
+        exported = pd.read_csv(fname, index_col=0, comment='#')
         exported.columns = self.reader._full_df.columns
         self.assertFrameEqual(exported, self.straditizer.full_df)
 

diff --git a/tests/widgets/test_straditizer_widgets.py b/tests/widgets/test_straditizer_widgets.py
@@ -13,6 +13,7 @@ def test_edit_attrs(self):
         n = len(self.window.dataframeeditors)
         editor, combo = self.straditizer_widgets.edit_attrs()
         self.assertEqual(len(self.window.dataframeeditors), n+1)
+        combo.addItem('latitude')
         combo.setCurrentText('latitude')
         self.assertIn('latitude', self.straditizer.attrs.index)