Coverage for crateweb/nlp_classification/tests/factories.py: 98%

1""" 

2crate_anon/crateweb/nlp_classification/tests/factories.py 

3 

4=============================================================================== 

5 

6 Copyright (C) 2015, University of Cambridge, Department of Psychiatry. 

7 Created by Rudolf Cardinal (rnc1001@cam.ac.uk). 

8 

9 This file is part of CRATE. 

10 

11 CRATE is free software: you can redistribute it and/or modify 

12 it under the terms of the GNU General Public License as published by 

13 the Free Software Foundation, either version 3 of the License, or 

14 (at your option) any later version. 

15 

16 CRATE is distributed in the hope that it will be useful, 

17 but WITHOUT ANY WARRANTY; without even the implied warranty of 

18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

19 GNU General Public License for more details. 

20 

21 You should have received a copy of the GNU General Public License 

22 along with CRATE. If not, see <https://www.gnu.org/licenses/>. 

23 

24=============================================================================== 

25 

26**CRATE NLP classification test factories.** 

27 

28""" 

29 

30from typing import Any, TYPE_CHECKING 

31 

32import factory 

33 

34from django.conf import settings 

35 

36from crate_anon.crateweb.nlp_classification.models import ( 

37 Assignment, 

38 Column, 

39 Option, 

40 Question, 

41 SampleSpec, 

42 SourceRecord, 

43 TableDefinition, 

44 Task, 

45 UserAnswer, 

46) 

47 

48if TYPE_CHECKING: 

49 from factory.builder import Resolver 

50 

51 

52class TaskFactory(factory.django.DjangoModelFactory): 

53 class Meta: 

54 model = Task 

55 

56 

57class TableDefinitionFactory(factory.django.DjangoModelFactory): 

58 class Meta: 

59 model = TableDefinition 

60 

61 

62class ColumnFactory(factory.django.DjangoModelFactory): 

63 class Meta: 

64 model = Column 

65 

66 table_definition = factory.SubFactory(TableDefinitionFactory) 

67 

68 

69class SourceRecordFactory(factory.django.DjangoModelFactory): 

70 class Meta: 

71 model = SourceRecord 

72 

73 source_column = factory.SubFactory(ColumnFactory) 

74 nlp_table_definition = factory.SubFactory(TableDefinitionFactory) 

75 

76 

77class SampleSpecFactory(factory.django.DjangoModelFactory): 

78 class Meta: 

79 model = SampleSpec 

80 

81 source_column = factory.SubFactory(ColumnFactory) 

82 nlp_table_definition = factory.SubFactory(TableDefinitionFactory) 

83 seed = factory.Faker("pyint", min_value=0, max_value=2147483647) 

84 size = factory.Faker("pyint", min_value=100, max_value=1000) 

85 

86 

87class UserFactory(factory.django.DjangoModelFactory): 

88 class Meta: 

89 model = settings.AUTH_USER_MODEL 

90 

91 username = factory.Sequence(lambda n: f"User {n+1}") 

92 

93 

94class AssignmentFactory(factory.django.DjangoModelFactory): 

95 class Meta: 

96 model = Assignment 

97 

98 task = factory.SubFactory(TaskFactory) 

99 sample_spec = factory.SubFactory(SampleSpecFactory) 

100 user = factory.SubFactory(UserFactory) 

101 

102 

103class QuestionFactory(factory.django.DjangoModelFactory): 

104 class Meta: 

105 model = Question 

106 skip_postgeneration_save = True 

107 

108 task = factory.SubFactory(TaskFactory) 

109 

110 @factory.post_generation 

111 def options( 

112 obj: "Resolver", create: bool, extracted: list[Option], **kwargs: Any 

113 ) -> None: 

114 if create and extracted: 

115 obj.options.add(*extracted) 

116 

117 

118class OptionFactory(factory.django.DjangoModelFactory): 

119 class Meta: 

120 model = Option 

121 

122 

123class UserAnswerFactory(factory.django.DjangoModelFactory): 

124 class Meta: 

125 model = UserAnswer 

126 

127 source_record = factory.SubFactory(SourceRecordFactory) 

128 question = factory.SubFactory(QuestionFactory) 

129 decision = factory.SubFactory(OptionFactory) 

130 assignment = factory.SubFactory(AssignmentFactory)