def test_load_with_filter(self):
fs = ['sepal_length', 'petal_width']
f, l = sql_data.load(self.db, 'SELECT * FROM iris.train LIMIT 3',
'class', fs)
self.assertEqual(len(fs), len(f))
self.assertEqual(3, len(l)) # label column length
def test_feature_columns(self):
f, l = sql_data.load(self.db, 'SELECT * FROM iris.train LIMIT 3',
'class', None)
c = sql_data.feature_columns(f)
self.assertEqual(4, len(c)) # 4 features
def test_load(self):
f, l = sql_data.load(self.db, 'SELECT * FROM iris.train LIMIT 3',
'class', None)
self.assertEqual(4, len(f.keys())) # 4 features
self.assertEqual(3, len(l)) # label column length
