Added test_odf

Author: WillAyd

pandas/tests/io/excel/test_odf.py

@@ -1,19 +1,28 @@
-odf = pytest.importorskip("odf")
+from collections import OrderedDict
 
+import numpy as np
+import pytest
 
-def test_get_sheet(self):
+import pandas as pd
+import pandas.util.testing as tm
+
+pytest.importorskip("odf")
+
+
+def test_get_sheet(datapath):
     from pandas.io.excel._odfreader import ODFReader
 
-    pth = os.path.join(self.dirpath, 'datatypes.ods')
+    pth = datapath("io", "data", "datatypes.ods")
     book = ODFReader(pth)
 
     assert len(book.sheet_names) == 1
     assert book.sheet_names == ['Sheet1']
 
-def test_get_sheet_raises(self):
+
+def test_get_sheet_raises(datapath):
     from pandas.io.excel._odfreader import ODFReader
 
-    pth = os.path.join(self.dirpath, 'datatypes.ods')
+    pth = datapath("io", "data", 'datatypes.ods')
     book = ODFReader(pth)
 
     with pytest.raises(ValueError):
@@ -25,11 +34,12 @@ def test_get_sheet_raises(self):
     with pytest.raises(IndexError):
         book.get_sheet_by_index(-33)
 
-def test_read_types(self):
-    sheet = self.get_exceldf(
-        'datatypes', '.ods', header=None, engine='odf')
 
-    expected = DataFrame(
+def test_read_types(datapath):
+    path = datapath("io", "data", "datatypes.ods")
+    sheet = pd.read_excel(path, header=None, engine='odf')
+
+    expected = pd.DataFrame(
         [[1.0],
          [1.25],
          ['a'],
@@ -43,32 +53,34 @@ def test_read_types(self):
          ['UBERON:0002101']])
     tm.assert_equal(sheet, expected)
 
-def test_read_invalid_types_raises(self):
+
+def test_read_invalid_types_raises(datapath):
     # the invalid_value_type.ods required manually editing
     # of the included content.xml file
+    path = datapath("io", "data", "invalid_value_type.ods")
     with pytest.raises(ValueError,
                        match="Unrecognized type awesome_new_type"):
-        self.get_exceldf(
-            'invalid_value_type', '.ods', header=None, engine='odf')
+        pd.read_excel(path, header=None, engine='odf')
 
-def test_read_lower_diagonal(self):
+
+def test_read_lower_diagonal(datapath):
     # Make sure we can parse:
     # 1
     # 2 3
     # 4 5 6
     # 7 8 9 10
-
-    sheet = self.get_exceldf(
-        'lowerdiagonal', '.ods', 'Sheet1',
-        index_col=None, header=None, engine='odf')
+    path = datapath("io", "data", "lowerdiagonal.ods")
+    sheet = pd.read_excel(path, 'Sheet1',
+                          index_col=None, header=None, engine='odf')
 
     assert sheet.shape == (4, 4)
 
-def test_read_headers(self):
-    sheet = self.get_exceldf(
-        'headers', '.ods', 'Sheet1', index_col=0, engine='odf')
 
-    expected = DataFrame.from_dict(OrderedDict([
+def test_read_headers(datapath):
+    path = datapath("io", "data", "headers.ods")
+    sheet = pd.read_excel(path, 'Sheet1', index_col=0, engine='odf')
+
+    expected = pd.DataFrame.from_dict(OrderedDict([
         ("Header", ["Row 1", "Row 2"]),
         ("Column 1", [1.0, 2.0]),
         ("Column 2", [3.0, 4.0]),
@@ -84,18 +96,19 @@ def test_read_headers(self):
         for value in sheet[name]:
             assert pd.isnull(value)
 
-def test_read_writer_table(self):
+
+def test_read_writer_table(datapath):
     # Also test reading tables from an text OpenDocument file
     # (.odt)
 
-    table = self.get_exceldf(
-        'writertable', '.odt', 'Table1', index_col=0, engine='odf')
+    path = datapath("io", "data", "writertable.odt")
+    table = pd.read_excel(path, 'Table1', index_col=0, engine='odf')
 
     assert table.shape == (3, 3)
-    expected = DataFrame.from_dict(OrderedDict([
+    expected = pd.DataFrame.from_dict(OrderedDict([
         ("Header", ["Row 1", "Row 2", "Row 3"]),
         ("Column 1", [1.0, 2.0, 3.0]),
-        ("Unnamed: 2", [nan, nan, nan]),
+        ("Unnamed: 2", [np.nan, np.nan, np.nan]),
         ("Column 3", [7.0, 8.0, 9.0])]))
     expected.set_index("Header", inplace=True)
     columns = ["Column 1", "Column 3"]
@@ -105,18 +118,20 @@ def test_read_writer_table(self):
     for i in range(3):
         assert pd.isnull(table["Unnamed: 2"][i])
 
-def test_blank_row_repeat(self):
-    table = self.get_exceldf(
-        'blank-row-repeat', '.ods', 'Value', engine='odf')
+
+def test_blank_row_repeat(datapath):
+    path = datapath("io", "data", "blank-row-repeat.ods")
+    table = pd.read_excel(path, 'Value', engine='odf')
 
     assert table.shape == (14, 2)
     assert table['value'][7] == 9.0
     assert pd.isnull(table['value'][8])
     assert not pd.isnull(table['value'][11])
 
-def test_runlengthencoding(self):
-    sheet = self.get_exceldf(
-        'runlengthencoding', '.ods', 'Sheet1', header=None, engine='odf')
+
+def test_runlengthencoding(datapath):
+    path = datapath("io", "data", "runlengthencoding.ods")
+    sheet = pd.read_excel(path, 'Sheet1', header=None, engine='odf')
     assert sheet.shape == (5, 3)
     # check by column, not by row.
     assert list(sheet[0]) == [1.0, 1.0, 2.0, 2.0, 2.0]