import pytest

import dbconnection.db as dbconnection


@pytest.fixture
def db():
    database = dbconnection.DB()
    database.connect()
    return database

def test_hash_partition(db:dbconnection.DB):
    db.query_file('../tidb-pcta-course-ja-lab/scripts/04-demo-hash-partition-setup.sql')
    assert True

def test_list_partition(db:dbconnection.DB):
    db.query_file('../tidb-pcta-course-ja-lab/scripts/04-demo-list-partition-setup.sql')
    assert True

def test_range_partition(db:dbconnection.DB):
    db.query_file('../tidb-pcta-course-ja-lab/scripts/04-demo-range-partition-setup.sql')
    assert True