Replace requirements.txt with pyproject.toml, add tests for models and update development documentation

Author: brayniverse

DEVELOPMENT.md

@@ -19,7 +19,7 @@ This document provides information for developers who want to contribute to the
 
 2. Install dependencies:
    ```bash
-   pip install -r requirements.txt
+   pip install -e .
    ```
 
 3. Set up environment variables as described in the `README.md`.
@@ -53,3 +53,17 @@ If migrations fail due to the `data/` directory missing, ensure it exists:
 ```bash
 mkdir -p data
 ```
+
+## Running Tests
+
+Testing is highly recommended for validating model relationships and ensuring data integrity.
+
+1. Install development dependencies:
+   ```bash
+   pip install -e ".[dev]"
+   ```
+
+2. Run tests using `pytest`:
+   ```bash
+   pytest
+   ```

pyproject.toml

@@ -0,0 +1,27 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "open-cinema-index"
+version = "0.1.0"
+description = "A movie index database"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "alembic",
+    "sqlalchemy",
+    "typer",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest",
+]
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.pytest.ini_options]
+pythonpath = ["src"]
+testpaths = ["tests"]

requirements.txt

@@ -0,0 +1,153 @@
+import pytest
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from open_cinema_index.models.base import Base
+from open_cinema_index.models.movie import Movie
+from open_cinema_index.models.person import Person
+from open_cinema_index.models.credit import Credit
+from open_cinema_index.models.attribute import Attribute, MovieAttribute
+from open_cinema_index.models.award import AwardCeremony, Award, MovieAward
+from open_cinema_index.models.movie import MovieAttributeUpdate
+
+@pytest.fixture
+def session():
+    # Use an in-memory SQLite database for testing
+    engine = create_engine("sqlite:///:memory:")
+    Base.metadata.create_all(engine)
+    with sessionmaker(bind=engine)() as session:
+        yield session
+    session.close()
+
+def test_create_movie(session):
+    movie = Movie(title="Inception", release_year=2010)
+    session.add(movie)
+    session.commit()
+    
+    saved_movie = session.query(Movie).filter_by(title="Inception").first()
+    assert saved_movie is not None
+    assert saved_movie.title == "Inception"
+    assert saved_movie.release_year == 2010
+
+def test_create_person(session):
+    person = Person(name="Christopher Nolan")
+    session.add(person)
+    session.commit()
+    
+    saved_person = session.query(Person).filter_by(name="Christopher Nolan").first()
+    assert saved_person is not None
+    assert saved_person.name == "Christopher Nolan"
+
+def test_movie_credit_relationship(session):
+    movie = Movie(title="Interstellar", release_year=2014)
+    person = Person(name="Matthew McConaughey")
+    session.add(movie)
+    session.add(person)
+    session.commit()
+    
+    credit = Credit(movie_id=movie.id, person_id=person.id, role="actor", character_name="Cooper")
+    session.add(credit)
+    session.commit()
+    
+    assert len(movie.credits) == 1
+    assert movie.credits[0].person.name == "Matthew McConaughey"
+    assert person.credits[0].movie.title == "Interstellar"
+
+def test_movie_attribute_relationship(session):
+    movie = Movie(title="The Matrix", release_year=1999)
+    attribute = Attribute(attribute_type="genre", value="Sci-Fi")
+    session.add(movie)
+    session.add(attribute)
+    session.commit()
+
+    movie_attr = MovieAttribute(movie_id=movie.id, attribute_id=attribute.id)
+    session.add(movie_attr)
+    session.commit()
+
+    assert len(movie.attributes) == 1
+    assert movie.attributes[0].attribute.value == "Sci-Fi"
+    assert len(attribute.movie_links) == 1
+    assert attribute.movie_links[0].movie.title == "The Matrix"
+
+def test_award_relationships(session):
+    ceremony = AwardCeremony(ceremony_name="Academy Awards")
+    award = Award(award_name="Best Picture", ceremony=ceremony)
+    movie = Movie(title="Parasite", release_year=2019)
+    person = Person(name="Bong Joon-ho")
+    
+    session.add_all([ceremony, award, movie, person])
+    session.commit()
+
+    movie_award = MovieAward(
+        movie_id=movie.id,
+        person_id=person.id,
+        award_id=award.id,
+        status="won",
+        year=2020
+    )
+    session.add(movie_award)
+    session.commit()
+
+    assert len(movie.awards) == 1
+    assert movie.awards[0].award.award_name == "Best Picture"
+    assert movie.awards[0].person.name == "Bong Joon-ho"
+    assert len(ceremony.awards) == 1
+    assert ceremony.awards[0].award_name == "Best Picture"
+
+def test_movie_attribute_update(session):
+    movie = Movie(title="Tenet", release_year=2020)
+    session.add(movie)
+    session.commit()
+
+    update = MovieAttributeUpdate(movie_id=movie.id, attribute_type="genre")
+    session.add(update)
+    session.commit()
+
+    assert len(movie.attribute_updates) == 1
+    assert movie.attribute_updates[0].attribute_type == "genre"
+
+def test_attribute_unique_constraint(session):
+    from sqlalchemy.exc import IntegrityError
+    attr1 = Attribute(attribute_type="color", value="Red")
+    session.add(attr1)
+    session.commit()
+
+    attr2 = Attribute(attribute_type="color", value="Red")
+    session.add(attr2)
+    with pytest.raises(IntegrityError):
+        session.commit()
+
+def test_cascade_delete_movie(session):
+    movie = Movie(title="Inception", release_year=2010)
+    person = Person(name="Leonardo DiCaprio")
+    session.add_all([movie, person])
+    session.commit()
+
+    credit = Credit(movie_id=movie.id, person_id=person.id, role="actor")
+    session.add(credit)
+    session.commit()
+
+    assert session.query(Credit).count() == 1
+    
+    session.delete(movie)
+    session.commit()
+    
+    assert session.query(Credit).count() == 0
+    assert session.query(Person).count() == 1
+
+def test_cascade_delete_person(session):
+    movie = Movie(title="Inception", release_year=2010)
+    person = Person(name="Leonardo DiCaprio")
+    session.add_all([movie, person])
+    session.commit()
+
+    credit = Credit(movie_id=movie.id, person_id=person.id, role="actor")
+    session.add(credit)
+    session.commit()
+
+    assert session.query(Credit).count() == 1
+    
+    session.delete(person)
+    session.commit()
+    
+    assert session.query(Credit).count() == 0
+    assert session.query(Movie).count() == 1