Skip to content

Commit

Permalink
Added more bq-tf tests
Browse files Browse the repository at this point in the history
  • Loading branch information
bovard committed Mar 2, 2021
1 parent d31a27f commit 25beb81
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/test_tensorflow_bigquery.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ class TestTensorflowBigQuery(unittest.TestCase):
# Some versions of bigquery crashed tensorflow, add this test to make sure that doesn't happen.
# python -c "from google.cloud import bigquery; import tensorflow". This flow is common because bigquery is imported in kaggle_gcp.py
# which is loaded at startup.
def test_addition(self):
def test_matrix_addition(self):
result = tf.add([1, 2], [3, 4])
self.assertEqual([2], result.shape)
self.assertListEqual([4, 6], [x for x in result.numpy()])

0 comments on commit 25beb81

Please sign in to comment.