138 lines
3.6 KiB
Python
138 lines
3.6 KiB
Python
"""Tests for core read_ods function with different files"""
|
|
from pathlib import Path
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
|
|
from pandas_ods_reader import read_ods
|
|
|
|
|
|
root = Path(__file__).parent
|
|
rsc = root / "rsc"
|
|
|
|
header_file = "example_headers.ods"
|
|
no_header_file = "example_no_headers.ods"
|
|
duplicated_column_names_file = "example_duplicated_column_names.ods"
|
|
col_len_file = "example_col_lengths.ods"
|
|
missing_header_file = "example_missing_header.ods"
|
|
mixed_dtypes_file = "mixed_dtypes.ods"
|
|
|
|
|
|
class TestOdsReader:
|
|
|
|
def test_header_file_simple(self):
|
|
|
|
path = rsc / header_file
|
|
df = read_ods(path)
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
def test_header_file_with_int(self):
|
|
|
|
path = rsc / header_file
|
|
df = read_ods(path, 1)
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
def test_header_file_with_str(self):
|
|
|
|
path = rsc / header_file
|
|
df = read_ods(path, "Sheet1")
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
def test_header_file_with_cols(self):
|
|
|
|
path = rsc / header_file
|
|
columns = ["One", "Two", "Three", "Four", "Five"]
|
|
df = read_ods(path, "Sheet1", columns=columns)
|
|
|
|
assert list(df.columns) == columns
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
def test_no_header_file_no_cols(self):
|
|
|
|
path = rsc / no_header_file
|
|
df = read_ods(path, 1, headers=False)
|
|
|
|
assert list(df.columns) == [
|
|
f"column.{i}" for i in range(len(df.columns))]
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
def test_no_header_file_with_cols(self):
|
|
|
|
path = rsc / no_header_file
|
|
columns = ["A", "B", "C", "D", "E"]
|
|
df = read_ods(path, 1, headers=False, columns=columns)
|
|
|
|
assert list(df.columns) == columns
|
|
assert len(df) == 10
|
|
|
|
def test_duplicated_column_names(self):
|
|
|
|
path = rsc / duplicated_column_names_file
|
|
df = read_ods(path, 1)
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert len(df.columns) == 4
|
|
assert "website.1" in df.columns
|
|
|
|
def test_header_file_col_len(self):
|
|
|
|
path = rsc / col_len_file
|
|
df = read_ods(path, 1)
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
def test_wrong_id_type(self):
|
|
|
|
path = rsc / header_file
|
|
|
|
with pytest.raises(ValueError) as e_info:
|
|
read_ods(path, 1.0)
|
|
assert e_info.match("Sheet id has to be either `str` or `int`")
|
|
|
|
def test_non_existent_sheet(self):
|
|
|
|
path = rsc / header_file
|
|
sheet_name = "No_Sheet"
|
|
|
|
with pytest.raises(KeyError) as e_info:
|
|
read_ods(path, sheet_name)
|
|
assert e_info.match(f"There is no sheet named {sheet_name}")
|
|
|
|
def test_missing_header(self):
|
|
|
|
path = rsc / missing_header_file
|
|
df = read_ods(path, 1)
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
assert df.columns[2] == "unnamed.1"
|
|
|
|
def test_mixed_dtypes(self):
|
|
|
|
path = rsc / mixed_dtypes_file
|
|
df = read_ods(path, 1)
|
|
|
|
assert isinstance(df, pd.DataFrame)
|
|
assert len(df) == 10
|
|
assert (len(df.columns) == 5)
|
|
|
|
type_list = [float, object, float, float, object]
|
|
assert df.dtypes.tolist() == type_list
|
|
col_b_types = [type(v) for v in df.B.values]
|
|
assert str in col_b_types and float in col_b_types
|