Old engine for Continuous Time Bayesian Networks. Superseded by reCTBN. 🐍
https://github.com/madlabunimib/PyCTBN
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
602 B
21 lines
602 B
4 years ago
|
from io import StringIO
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
from .base import BaseExtensionTests
|
||
|
|
|
||
|
|
|
||
|
|
class BaseParsingTests(BaseExtensionTests):
|
||
|
|
@pytest.mark.parametrize("engine", ["c", "python"])
|
||
|
|
def test_EA_types(self, engine, data):
|
||
|
|
df = pd.DataFrame({"with_dtype": pd.Series(data, dtype=str(data.dtype))})
|
||
|
|
csv_output = df.to_csv(index=False, na_rep=np.nan)
|
||
|
|
result = pd.read_csv(
|
||
|
|
StringIO(csv_output), dtype={"with_dtype": str(data.dtype)}, engine=engine
|
||
|
|
)
|
||
|
|
expected = df
|
||
|
|
self.assert_frame_equal(result, expected)
|