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

    def test_uses_country_table(self):
        """Check if data comes from the country table."""
        # We'll check if all results are among the capitals from the country table
        with open("query.sql", "r") as file:
            query = file.read()
        # Get expected capitals from country
        self.cursor.execute("SELECT capital FROM country")
        expected_capitals = set(row[0] for row in self.cursor.fetchall())
        self.cursor.execute(query)
        result_capitals = set(row[0] for row in self.cursor.fetchall())
        condition = result_capitals.issubset(expected_capitals)
        _dynamic_test(
            self,
            condition,
            "Query retrieves capitals from the correct table.",
            "Query result includes values not in the 'country' table's 'capital' column."
        )

    def test_orders_by_capital_desc(self):
        """Check if results are sorted by capital in descending order."""
        with open("query.sql", "r") as file:
            query = file.read()
        self.cursor.execute(query)
        result = [row[0] for row in self.cursor.fetchall()]
        sorted_desc = sorted(result, reverse=True)
        condition = result == sorted_desc
        _dynamic_test(
            self,
            condition,
            "Results are sorted by 'capital' in descending order.",
            f"Results are not sorted in descending order: {result}"
        )

    def test_all_capitals_included(self):
        """Check if all capitals from country table are included in result (including NULLs if present)."""
        with open("query.sql", "r") as file:
            query = file.read()
        self.cursor.execute("SELECT capital FROM country")
        expected = set(row[0] for row in self.cursor.fetchall())
        self.cursor.execute(query)
        result = set(row[0] for row in self.cursor.fetchall())
        condition = expected == result
        _dynamic_test(
            self,
            condition,
            "All capitals from the country table are included in the result.",
            f"Missing or extra capitals in result. Expected: {expected}, Got: {result}"
        )

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

Erkunden Sie die Grundlagen von SQL, von der Abfrage und Filterung von Daten mit SELECT und WHERE bis hin zur Sortierung von Ergebnissen mit ORDER BY und deren Zusammenfassung mit Aggregatfunktionen wie COUNT, AVG und SUM. Durch praxisorientierte Aufgaben erwerben Sie die Fähigkeiten, reale Datensätze sicher abzufragen.

Erfahren Sie, wie Sie einzelne, mehrere und alle Spalten aus einer Tabelle mit der SQL-SELECT-Anweisung abrufen. Lernen Sie, wie Sie eindeutige Zeilen zurückgeben und die Anzahl der von einer Abfrage zurückgegebenen Ergebnisse begrenzen.

Erfahren Sie, wie Sie Abfrageergebnisse mit der ORDER BY-Klausel organisieren. Lernen Sie, Daten nach einer oder mehreren Spalten zu sortieren und die Sortierreihenfolge mit aufsteigender oder absteigender Ordnung für eine klarere Ausgabe zu steuern.



Erfahren Sie, wie Sie Abfrageergebnisse mit der WHERE-Klausel filtern. Gleichheits- und Ungleichheitsprüfungen, Überprüfung von Werten innerhalb eines Bereichs sowie Identifizierung von NULL-Werten in Ihren Daten.

Erfahren Sie, wie Sie Ihre Suchbedingungen verfeinern, indem Sie WHERE-Klauseln mit den Operatoren AND und OR kombinieren. Lernen Sie, den Operator NOT zu verwenden, um bestimmte Ergebnisse aus Ihrer Abfrage auszuschließen.



Erfahren Sie, wie Tabellendaten mithilfe von SQL-Aggregatfunktionen zusammengefasst werden. Lernen Sie, Durchschnittswerte und Summen mit AVG und SUM zu berechnen, Zeilen mit COUNT zu zählen und Extremwerte mit MAX und MIN zu ermitteln.



Herausforderung: Hauptstädte in Absteigender Reihenfolge Sortieren

Lösung