Commit 2a3db41
committed
Add datatable and tests
1 parent 06bf91f commit 2a3db41
3 files changed
Lines changed: 34 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -411,6 +411,7 @@ RUN pip install flashtext && \ | |||
| 411 | 411 | pip install pydub && \ | |
| 412 | 412 | pip install pydegensac && \ | |
| 413 | 413 | pip install pytorch-lightning && \ | |
| 414 | + pip install datatable && \ | ||
| 414 | 415 | /tmp/clean-layer.sh | |
| 415 | 416 | ||
| 416 | 417 | # Tesseract and some associated utility packages | |
| Original file line number | Diff line number | Diff 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 | ||
| Original file line number | Diff line number | Diff 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