experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix arithmetic transform test (#3134)

Browse Source
This commit is contained in:
Yuan (Terry) Tang 2016-07-01 10:37:05 -05:00 committed by Rasmus Munk Larsen
parent aa2cacd662
commit 084c915c2d
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/contrib/learn/python/learn/tests/dataframe/arithmetic_transform_test.py
View File

@ -20,16 +20,24 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import numpy as np import numpy as np
import pandas as pd
import tensorflow as tf import tensorflow as tf
from tensorflow.contrib.learn.python.learn.dataframe import tensorflow_dataframe as df from tensorflow.contrib.learn.python.learn.dataframe import tensorflow_dataframe as df
# pylint: disable=g-import-not-at-top
try:
import pandas as pd
HAS_PANDAS = True
except ImportError:
HAS_PANDAS = False
class SumTestCase(tf.test.TestCase): class SumTestCase(tf.test.TestCase):
"""Test class for `Sum` transform.""" """Test class for `Sum` transform."""
def testSum(self): def testSum(self):
if not HAS_PANDAS:
return
num_rows = 100 num_rows = 100
pandas_df = pd.DataFrame({"a": np.arange(num_rows), pandas_df = pd.DataFrame({"a": np.arange(num_rows),