diff --git a/Dockerfile b/Dockerfile index f411e644..c0d58c89 100644 --- a/Dockerfile +++ b/Dockerfile @@ -411,6 +411,7 @@ RUN pip install flashtext && \ pip install pydub && \ pip install pydegensac && \ pip install pytorch-lightning && \ + pip install datatable && \ /tmp/clean-layer.sh # Tesseract and some associated utility packages diff --git a/tests/test_datatable.py b/tests/test_datatable.py new file mode 100644 index 00000000..ec68ce12 --- /dev/null +++ b/tests/test_datatable.py @@ -0,0 +1,17 @@ +import unittest +import datatable as dt +from datatable.internal import frame_integrity_check +from datatable import ltype + +class TestDatatable(unittest.TestCase): + def test_fread(self): + d0 = dt.fread( + "L,T,U,D\n" + "true,True,TRUE,1\n" + "false,False,FALSE,0\n" + ",,,\n" + ) + frame_integrity_check(d0) + assert d0.shape == (3, 4) + assert d0.ltypes == (ltype.bool,) * 4 + assert d0.to_list() == [[True, False, None]] * 4