Skip to content

Commit 2a3db41

Browse files
committed
Add datatable and tests
1 parent 06bf91f commit 2a3db41

File tree

3 files changed

+34
-0
lines changed

3 files changed

+34
-0
lines changed

Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -411,6 +411,7 @@ RUN pip install flashtext && \
411411
pip install pydub && \
412412
pip install pydegensac && \
413413
pip install pytorch-lightning && \
414+
pip install datatable && \
414415
/tmp/clean-layer.sh
415416

416417
# Tesseract and some associated utility packages

tests/test_datatable.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
import unittest
2+
import datatable as dt
3+
from datatable.internal import frame_integrity_check
4+
from datatable import ltype
5+
6+
class TestDatatable(unittest.TestCase):
7+
def test_fread(self):
8+
d0 = dt.fread(
9+
"L,T,U,D\n"
10+
"true,True,TRUE,1\n"
11+
"false,False,FALSE,0\n"
12+
",,,\n"
13+
)
14+
frame_integrity_check(d0)
15+
assert d0.shape == (3, 4)
16+
assert d0.ltypes == (ltype.bool,) * 4
17+
assert d0.to_list() == [[True, False, None]] * 4

tests/test_datatable.py~

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
import unittest
2+
import datatable as dt
3+
from datatable.internal import frame_integrity_check
4+
5+
class TestDatatable(unittest.TestCase):
6+
def test_fread(self):
7+
d0 = dt.fread(
8+
"L,T,U,D\n"
9+
"true,True,TRUE,1\n"
10+
"false,False,FALSE,0\n"
11+
",,,\n"
12+
)
13+
frame_integrity_check(d0)
14+
assert d0.shape == (3, 4)
15+
assert d0.ltypes == (ltype.bool,) * 4
16+
assert d0.to_list() == [[True, False, None]] * 4

0 commit comments

Comments
 (0)