Add the vaex library (includes a test) (#1031)

Kaggle · Jun 18, 2021 · 411e2f5 · 411e2f5
1 parent 6d08285
commit 411e2f5
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -365,6 +365,7 @@ RUN pip install bleach && \
 
 RUN pip install flashtext && \
     pip install wandb && \
+    pip install vaex && \
     pip install marisa-trie && \
     pip install pyemd && \
     pip install pyupset && \

diff --git a/tests/test_vaex.py b/tests/test_vaex.py
@@ -0,0 +1,10 @@
+import unittest
+
+import vaex
+
+class TestVaex(unittest.TestCase):
+    def test_read_csv(self):
+        df = vaex.read_csv("/input/tests/data/train.csv")
+
+        self.assertEqual((100, 785), df.shape)
+        self.assertEqual(10, df['label'].nunique())