From 2a3db411ac460961e759d1af14790a25974bfa88 Mon Sep 17 00:00:00 2001 From: marketneutral Date: Tue, 1 Dec 2020 19:19:47 +0000 Subject: [PATCH 1/2] Add datatable and tests --- Dockerfile | 1 + tests/test_datatable.py | 17 +++++++++++++++++ tests/test_datatable.py~ | 16 ++++++++++++++++ 3 files changed, 34 insertions(+) create mode 100644 tests/test_datatable.py create mode 100644 tests/test_datatable.py~ 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 diff --git a/tests/test_datatable.py~ b/tests/test_datatable.py~ new file mode 100644 index 00000000..efbd2a3f --- /dev/null +++ b/tests/test_datatable.py~ @@ -0,0 +1,16 @@ +import unittest +import datatable as dt +from datatable.internal import frame_integrity_check + +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 From 4bcce7f1101102e199f0a30b394813702765f03d Mon Sep 17 00:00:00 2001 From: marketneutral Date: Tue, 1 Dec 2020 19:24:26 +0000 Subject: [PATCH 2/2] Remove old test file --- tests/test_datatable.py~ | 16 ---------------- 1 file changed, 16 deletions(-) delete mode 100644 tests/test_datatable.py~ diff --git a/tests/test_datatable.py~ b/tests/test_datatable.py~ deleted file mode 100644 index efbd2a3f..00000000 --- a/tests/test_datatable.py~ +++ /dev/null @@ -1,16 +0,0 @@ -import unittest -import datatable as dt -from datatable.internal import frame_integrity_check - -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