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_query_returns_4_rows(self):
        """Check that the query returns exactly 4 rows."""
        with open("query.sql", "r") as file:
            query = file.read()
        self.cursor.execute(query)
        rows = self.cursor.fetchall()
        _dynamic_test(
            self,
            len(rows) == 4,
            "Query returns exactly 4 rows.",
            f"Query should return 4 rows, got {len(rows)}."
        )

    def test_query_returns_unique_regions(self):
        """Check that all rows in the result have unique region values."""
        with open("query.sql", "r") as file:
            query = file.read()
        self.cursor.execute(query)
        rows = self.cursor.fetchall()
        regions = [row[0] for row in rows]
        unique_regions = set(regions)
        _dynamic_test(
            self,
            len(unique_regions) == len(regions),
            "All returned regions are unique.",
            f"Returned regions are not unique: {regions}."
        )

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

Explore os fundamentos do SQL — desde a consulta de dados até a integração em aplicações do mundo real. Seja você iniciante em gerenciamento de bancos de dados ou esteja 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 a consulta e manipulação de dados até a integração eficiente em aplicações reais. 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 descobriremos 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 classificar os resultados por uma ou várias colunas, em ordem crescente ou decrescente, proporcionando uma saída de dados mais significativa e de fácil interpretação.

Nesta seção, aprenderemos como filtrar dados retornados utilizando a cláusula WHERE da instrução SELECT. Serão abordados testes de igualdade, desigualdade, intervalos de valores maiores e menores, além de 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 de agregação do SQL e como utilizá-las para resumir dados de tabelas.

Desafio: Encontre as Regiões em Que Todos os Países Estão Localizados

Solução