from src.models import B4wExport, Category from tests.factories import B4wExportFactory def test_b4wexport_simple(): data = B4wExportFactory.build() assert isinstance(data, B4wExport) def test_b4export_category(): data = B4wExportFactory.build(Category="fun:stuff") assert isinstance(data.category, Category) assert data.category.to_str() == "fun:stuff"