def test_embedded(self) -> None: """Test embedded method to check JSON-serialization and full join.""" assert Recommendation.from_id(1).to_json(join=True) == { 'id': 1, 'group_id': 1, 'tag_id': 1, 'time': '2020-10-24 20:02:00' + ('' if config.backend == 'sqlite' else '-04:00'), 'priority': 1, 'object_id': 1, 'facility_id': 1, 'user_id': 2, 'forecast_id': 1, 'predicted_observation_id': 11, 'observation_id': 19, 'accepted': True, 'rejected': False, 'data': {}, 'group': RecommendationGroup.from_id(1).to_json(join=True), 'tag': RecommendationTag.from_id(1).to_json(join=True), 'user': User.from_id(2).to_json(join=True), 'facility': Facility.from_id(1).to_json(join=True), 'object': Object.from_id(1).to_json(join=True), 'forecast': Forecast.from_id(1).to_json(join=True), 'predicted': Observation.from_id(11).to_json(join=True), 'observed': Observation.from_id(19).to_json(join=True), }
def test_relationship_observed(self, testdata: TestData) -> None: """Test observation foreign key relationship on recommendation.""" for i, record in enumerate(testdata['recommendation']): assert Recommendation.from_id( i + 1).observed.id == record['observation_id']
def test_relationship_forecast(self, testdata: TestData) -> None: """Test forecast foreign key relationship on recommendation.""" for i, record in enumerate(testdata['recommendation']): assert Recommendation.from_id( i + 1).forecast.id == record['forecast_id']
def test_relationship_group(self, testdata: TestData) -> None: """Test recommendation_group foreign key relationship on recommendation.""" for i, record in enumerate(testdata['recommendation']): assert Recommendation.from_id(i + 1).group.id == record['group_id']
def test_id_missing(self) -> None: """Test exception on missing recommendation `id`.""" with pytest.raises(NotFound): Recommendation.from_id(-1)
def test_from_id(self, testdata: TestData) -> None: """Test loading recommendation from `id`.""" # NOTE: `id` not set until after insert for i, record in enumerate(testdata['recommendation']): assert Recommendation.from_id(i + 1).id == i + 1