def test_get_columns_specified(self): qb = QueryBuilder('company') columns = 'a, b' qb.select(columns) self.assertEqual(qb.get_columns(), columns)
def test_get_columns_specified(self): qb = QueryBuilder("company") columns = "a, b" qb.select(columns) self.assertEqual(qb.get_columns(), columns)
def test_get_columns_all(self): qb = QueryBuilder('company') self.assertEqual(qb.get_columns(), '*')
def test_get_columns_all(self): qb = QueryBuilder("company") self.assertEqual(qb.get_columns(), "*")