Source code for cis_interface.serialize.tests.test_PandasSerialize

import numpy as np
import pandas as pd
from cis_interface.tests import assert_raises
from cis_interface.serialize.tests import test_AsciiTableSerialize as parent


[docs]class TestPandasSerialize(parent.TestAsciiTableSerialize):
    r"""Test class for TestPandasSerialize class."""

    _cls = 'PandasSerialize'

[docs]    def test_apply_field_names_errors(self):
        r"""Test errors raised by apply_field_names."""
        assert_raises(RuntimeError, self.instance.apply_field_names,
                      pd.DataFrame({'x': np.zeros(3), 'y': np.zeros(3)}))
        names = self.testing_options['objects'][0].columns.tolist()
        names[0] = 'invalid'
        assert_raises(RuntimeError, self.instance.apply_field_names,
                      pd.DataFrame({k: np.zeros(3) for k in names}))

[docs]    def test_func_serialize_errors(self):
        r"""Test errors raised by func_serialize."""
        assert_raises(TypeError, self.instance.func_serialize, None)