import unittest
import psycopg2
import pandas as pd

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_name_column(self):
        """Check if the query selects only the 'name' column."""
        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 == ["name"]
        _dynamic_test(
            self,
            condition,
            "Query selects only the 'name' column.",
            f"Query should select only the 'name' column but got: {columns}"
        )

    def test_correct_rows_returned(self):
        """Check if query returns only countries from 'Central America' or 'North America' regions."""
        with open("query.sql", "r") as file:
            query = file.read()
        self.cursor.execute(query)
        rows = self.cursor.fetchall()
        result_names = sorted([r[0].strip() for r in rows])
        expected_names = sorted([
            'Mexico',
            'Paraguay',
        ])
        _dynamic_test(
            self,
            result_names == expected_names,
            f"Query returns only countries from 'Central America' or 'North America'.",
            f"Expected names: {expected_names}, got: {result_names}"
        )

    def test_or_operator_used(self):
        """Check that both 'Central America' and 'North America' regions are included in results."""
        with open("query.sql", "r") as file:
            query = file.read().lower().replace("\\n", " ")
        condition = ("or" in query) and ("central america" in query) and ("north america" in query)
        _dynamic_test(
            self,
            condition,
            "Query uses OR operator and includes both regions.",
            "Query must use OR operator and include both 'Central America' and 'North America' in WHERE clause."
        )

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

This course is perfect for beginners ready to explore the world of SQL. Whether you're just starting out in database management or aiming to use SQL for your application development projects, this course covers the essentials. You'll quickly learn how to leverage the full potential of SQL, from querying and managing data to seamlessly integrating it into real-world applications. By the end of the course, you'll have the confidence and skills to solve practical problems with SQL and enhance your development process.

Challenge: America Regions Filter

Solution