def test_add_queries_with_table_name(self):
query_a = Query('mock_table').where(col.numField > 4)
query_b = Query().take(5)
query = query_a + query_b
self.assertEqual(
"mock_table | where numField > 4 | take 5",
query.render(),
)
self.assertEqual(
"mock_table",
query.get_table_name(),
)
# make sure the originals didn't change
self.assertEqual(
"mock_table | where numField > 4",
query_a.render(),
)
self.assertEqual(
"mock_table",
query_a.get_table_name(),
)
self.assertEqual(
" | take 5",
query_b.render(),
)
self.assertEqual(
None,
query_b.get_table_name(),
)
