Skip to content

Commit 25beb81

Browse files
committed
Added more bq-tf tests
1 parent d31a27f commit 25beb81

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_tensorflow_bigquery.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ class TestTensorflowBigQuery(unittest.TestCase):
99
# Some versions of bigquery crashed tensorflow, add this test to make sure that doesn't happen.
1010
# python -c "from google.cloud import bigquery; import tensorflow". This flow is common because bigquery is imported in kaggle_gcp.py
1111
# which is loaded at startup.
12-
def test_addition(self):
12+
def test_matrix_addition(self):
1313
result = tf.add([1, 2], [3, 4])
1414
self.assertEqual([2], result.shape)
15+
self.assertListEqual([4, 6], [x for x in result.numpy()])

0 commit comments

Comments
 (0)