Fix EtlTest test able to run every where (#97)

databricks · Apr 19, 2019 · 2a3be45 · 2a3be45
1 parent 9479fbb
commit 2a3be45
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/databricks/koalas/tests/test_etl.py b/databricks/koalas/tests/test_etl.py
@@ -14,6 +14,7 @@
 # limitations under the License.
 #
 
+import os
 import unittest
 
 import pandas as pd
@@ -25,7 +26,8 @@ class EtlTest(ComparisonTestBase):
 
     @property
     def pdf(self):
-        return pd.read_csv('data/sample_stocks.csv')
+        test_dir = os.path.dirname(os.path.realpath(__file__))
+        return pd.read_csv('%s/../../../data/sample_stocks.csv' % test_dir)
 
     @compare_both
     def test_etl(self, df):