docker-python/tests/test_tensorflow_bigquery.py at main · LumaKernel/docker-python · GitHub

15 lines (11 loc) · 584 Bytes

import unittest

from google.cloud import bigquery

import tensorflow as tf

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_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()])

Expand file tree