import factory
from notes.models import ChapterBookmark, ModuleBookmark


class ChapterBookmarkFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = ChapterBookmark

class ModuleBookmarkFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = ModuleBookmark