def test_scalar_to_query_parameter_w_special_floats(self):
nan_parameter = _helpers.scalar_to_query_parameter(float("nan"))
self.assertTrue(math.isnan(nan_parameter.value))
self.assertEqual(nan_parameter.type_, "FLOAT64")
inf_parameter = _helpers.scalar_to_query_parameter(float("inf"))
self.assertTrue(math.isinf(inf_parameter.value))
self.assertEqual(inf_parameter.type_, "FLOAT64")
def test_scalar_to_query_parameter(self):
expected_types = [
(True, "BOOL"),
(False, "BOOL"),
(123, "INT64"),
(-123456789, "INT64"),
(1.25, "FLOAT64"),
(decimal.Decimal("1.25"), "NUMERIC"),
(b"I am some bytes", "BYTES"),
(u"I am a string", "STRING"),
(datetime.date(2017, 4, 1), "DATE"),
(datetime.time(12, 34, 56), "TIME"),
(datetime.datetime(2012, 3, 4, 5, 6, 7), "DATETIME"),
(
datetime.datetime(
2012, 3, 4, 5, 6, 7, tzinfo=google.cloud._helpers.UTC
),
"TIMESTAMP",
),
]
for value, expected_type in expected_types:
msg = "value: {} expected_type: {}".format(value, expected_type)
parameter = _helpers.scalar_to_query_parameter(value)
self.assertIsNone(parameter.name, msg=msg)
self.assertEqual(parameter.type_, expected_type, msg=msg)
self.assertEqual(parameter.value, value, msg=msg)
named_parameter = _helpers.scalar_to_query_parameter(value, name="myvar")
self.assertEqual(named_parameter.name, "myvar", msg=msg)
self.assertEqual(named_parameter.type_, expected_type, msg=msg)
self.assertEqual(named_parameter.value, value, msg=msg)
def test_scalar_to_query_parameter_w_special_floats(self):
nan_parameter = _helpers.scalar_to_query_parameter(float('nan'))
self.assertTrue(math.isnan(nan_parameter.value))
self.assertEqual(nan_parameter.type_, 'FLOAT64')
inf_parameter = _helpers.scalar_to_query_parameter(float('inf'))
self.assertTrue(math.isinf(inf_parameter.value))
self.assertEqual(inf_parameter.type_, 'FLOAT64')
def test_scalar_to_query_parameter(self):
expected_types = [
(True, 'BOOL'),
(False, 'BOOL'),
(123, 'INT64'),
(-123456789, 'INT64'),
(1.25, 'FLOAT64'),
(b'I am some bytes', 'BYTES'),
(u'I am a string', 'STRING'),
(datetime.date(2017, 4, 1), 'DATE'),
(datetime.time(12, 34, 56), 'TIME'),
(datetime.datetime(2012, 3, 4, 5, 6, 7), 'DATETIME'),
(
datetime.datetime(2012,
3,
4,
5,
6,
7,
tzinfo=google.cloud._helpers.UTC),
'TIMESTAMP',
),
]
for value, expected_type in expected_types:
msg = 'value: {} expected_type: {}'.format(value, expected_type)
parameter = _helpers.scalar_to_query_parameter(value)
self.assertIsNone(parameter.name, msg=msg)
self.assertEqual(parameter.type_, expected_type, msg=msg)
self.assertEqual(parameter.value, value, msg=msg)
named_parameter = _helpers.scalar_to_query_parameter(value,
name='myvar')
self.assertEqual(named_parameter.name, 'myvar', msg=msg)
self.assertEqual(named_parameter.type_, expected_type, msg=msg)
self.assertEqual(named_parameter.value, value, msg=msg)
def test_scalar_to_query_parameter(self):
expected_types = [
(True, 'BOOL'),
(False, 'BOOL'),
(123, 'INT64'),
(-123456789, 'INT64'),
(1.25, 'FLOAT64'),
(b'I am some bytes', 'BYTES'),
(u'I am a string', 'STRING'),
(datetime.date(2017, 4, 1), 'DATE'),
(datetime.time(12, 34, 56), 'TIME'),
(datetime.datetime(2012, 3, 4, 5, 6, 7), 'DATETIME'),
(
datetime.datetime(
2012, 3, 4, 5, 6, 7, tzinfo=google.cloud._helpers.UTC),
'TIMESTAMP',
),
]
for value, expected_type in expected_types:
msg = 'value: {} expected_type: {}'.format(value, expected_type)
parameter = _helpers.scalar_to_query_parameter(value)
self.assertIsNone(parameter.name, msg=msg)
self.assertEqual(parameter.type_, expected_type, msg=msg)
self.assertEqual(parameter.value, value, msg=msg)
named_parameter = _helpers.scalar_to_query_parameter(
value, name='myvar')
self.assertEqual(named_parameter.name, 'myvar', msg=msg)
self.assertEqual(named_parameter.type_, expected_type, msg=msg)
self.assertEqual(named_parameter.value, value, msg=msg)
def test_scalar_to_query_parameter_honors_given_type_errors_on_invalid():
with pytest.raises(
google.cloud.bigquery.dbapi.exceptions.ProgrammingError,
match=
"The given parameter type, INT, for foo is not a valid BigQuery scalar type.",
):
_helpers.scalar_to_query_parameter(None, "foo", "INT")
def test_scalar_to_query_parameter(self):
expected_types = [
(True, "BOOL"),
(False, "BOOL"),
(123, "INT64"),
(-123456789, "INT64"),
(1.25, "FLOAT64"),
(decimal.Decimal("1.25"), "NUMERIC"),
(b"I am some bytes", "BYTES"),
(u"I am a string", "STRING"),
(datetime.date(2017, 4, 1), "DATE"),
(datetime.time(12, 34, 56), "TIME"),
(datetime.datetime(2012, 3, 4, 5, 6, 7), "DATETIME"),
(
datetime.datetime(
2012, 3, 4, 5, 6, 7, tzinfo=google.cloud._helpers.UTC
),
"TIMESTAMP",
),
]
for value, expected_type in expected_types:
msg = "value: {} expected_type: {}".format(value, expected_type)
parameter = _helpers.scalar_to_query_parameter(value)
self.assertIsNone(parameter.name, msg=msg)
self.assertEqual(parameter.type_, expected_type, msg=msg)
self.assertEqual(parameter.value, value, msg=msg)
named_parameter = _helpers.scalar_to_query_parameter(value, name="myvar")
self.assertEqual(named_parameter.name, "myvar", msg=msg)
self.assertEqual(named_parameter.type_, expected_type, msg=msg)
self.assertEqual(named_parameter.value, value, msg=msg)
def test_scalar_to_query_parameter_honors_given_type(alias, type_):
from google.cloud import bigquery
assert _helpers.scalar_to_query_parameter(
1.23, None,
alias) == (bigquery.ScalarQueryParameter(None, type_, 1.23))
assert _helpers.scalar_to_query_parameter(
None, "foo",
alias) == (bigquery.ScalarQueryParameter("foo", type_, None))
def test_scalar_to_query_parameter_w_unexpected_type(self):
with self.assertRaises(exceptions.ProgrammingError):
_helpers.scalar_to_query_parameter(value={"a": "dictionary"})
def test_scalar_to_query_parameter_w_unexpected_type(self):
with self.assertRaises(exceptions.ProgrammingError):
_helpers.scalar_to_query_parameter(value={'a': 'dictionary'})
