import unittest
import psycopg2
import pandas as pd
import re

class TestPostgresQuery(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.connection = psycopg2.connect(
            host="localhost",
            user="agent",
            password="qwerty",
            dbname="testdb",
            sslmode="disable"
        )
        cls.cursor = cls.connection.cursor()

    def test_selects_population_column_only(self):
        """Check that only the 'population' column is selected from country table."""
        with open("query.sql", "r") as file:
            query = file.read()
        self.cursor.execute(query)
        columns = [desc[0] for desc in self.cursor.description]
        condition = columns == ["population"]
        _dynamic_test(
            self,
            condition,
            "Query selects only the 'population' column.",
            f"Query must select only the 'population' column, but columns returned are: {columns}"
        )

    def test_population_values_correct(self):
        """Check that all returned population values match those in the table."""
        with open("query.sql", "r") as file:
            query = file.read()
        self.cursor.execute(query)
        user_rows = self.cursor.fetchall()
        self.cursor.execute("SELECT population FROM country;")
        expected_rows = self.cursor.fetchall()
        user_df = pd.DataFrame(user_rows, columns=["population"])
        expected_df = pd.DataFrame(expected_rows, columns=["population"])
        user_sorted = user_df.sort_values(by=["population"], na_position='last').reset_index(drop=True)
        expected_sorted = expected_df.sort_values(by=["population"], na_position='last').reset_index(drop=True)
        condition = user_sorted.equals(expected_sorted)
        if not condition:
            diff = pd.concat([expected_sorted, user_sorted]).drop_duplicates(keep=False)
            failure_msg = f"Returned population values do not match expected values. Difference:\\n{diff}"
        else:
            failure_msg = ""
        _dynamic_test(
            self,
            condition,
            "All population values returned correctly.",
            failure_msg
        )

def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)

def normalize_text(text):
    text = text.lower()
    text = re.sub(r"\s{2,}", " ", text)
    text = re.sub(r"\s*([,:?])\s*", r"\1 ", text)
    return text.strip()

if __name__ == "__main__":
    unittest.main()


test_solution.py

Este curso é ideal para iniciantes que desejam explorar o mundo do SQL. Seja você alguém começando em gerenciamento de bancos de dados ou buscando utilizar SQL em projetos de desenvolvimento de aplicações, este curso aborda os conceitos essenciais. Você aprenderá rapidamente a aproveitar todo o potencial do SQL, desde consultas e gerenciamento de dados até a integração eficiente em aplicações do mundo real. Ao final do curso, você terá confiança e habilidades para resolver problemas práticos com SQL e aprimorar seu processo de desenvolvimento.

Nesta seção, aprenderemos como utilizar a instrução SELECT do SQL para obter uma coluna de tabela, múltiplas colunas de tabela e todas as colunas de tabela. Também veremos como retornar valores exclusivos.

Nesta seção, exploraremos como organizar e priorizar os dados retornados por uma consulta utilizando a cláusula ORDER BY. Inserida ao final de uma instrução SELECT, essa cláusula permite ordenar os resultados por uma ou várias colunas, em ordem ascendente ou descendente, proporcionando uma saída de dados mais significativa e de fácil interpretação.

Nesta seção, aprenderemos como filtrar os dados retornados utilizando a cláusula WHERE da instrução SELECT. Aprenderemos a testar igualdade, desigualdade, valores maiores e menores dentro de intervalos, e valores NULL.

Nesta seção, exploraremos como refinar suas condições de busca combinando cláusulas WHERE com os operadores AND e OR. Também abordaremos como o operador NOT pode ser utilizado para excluir determinados resultados.

Nesta seção, aprenderemos o que são as funções agregadas do SQL e como utilizá-las para resumir dados de tabelas.

Desafio: Encontrar a População dos Países

Solução