Use absolute import path

tests/base.py

@@ -1,18 +1,21 @@
-import unittest
 import tensorflow as tf
+import unittest
+
 
 def train_input_fn(features, labels, batch_size=32):
     dataset = tf.data.Dataset.from_tensor_slices((dict(features), labels))
     dataset = dataset.shuffle(1000).repeat().batch(batch_size)
     return dataset
 
+
 def eval_input_fn(features, labels, batch_size=32):
     dataset = tf.data.Dataset.from_tensor_slices((dict(features), labels))
     dataset = dataset.batch(batch_size)
     return dataset
 
+
 class BaseTestCases:
-    class BaseTest(object):
+    class BaseTest(unittest.TestCase):
         def setUp(self):
             self.model, self.features, self.label = None, {}, None
 

tests/test_dnnclassifier.py

@@ -1,5 +1,5 @@
 import sqlflow_models
-from .base import BaseTestCases
+from tests.base import BaseTestCases
 
 import tensorflow as tf
 import unittest

tests/test_lstm.py

@@ -1,5 +1,5 @@
 import sqlflow_models
-from .base import BaseTestCases
+from tests.base import BaseTestCases
 
 import tensorflow as tf
 import numpy as np

tests/test_version.py

@@ -1,5 +1,4 @@
 import sqlflow_models
-import unittest
 
 
 def test_answer():