Add a decimal attribute in code structure tutorial and test

robcxyz · robcxyz · commit 3bb988b002fa · 2021-09-18T22:20:17.000-07:00
diff --git a/docs_src/tutorial/code_structure/tutorial001/app.py b/docs_src/tutorial/code_structure/tutorial001/app.py
@@ -9,7 +9,7 @@ def create_heroes():
         team_z_force = Team(name="Z-Force", headquarters="Sister Margaret’s Bar")
 
         hero_deadpond = Hero(
-            name="Deadpond", secret_name="Dive Wilson", team=team_z_force
+            name="Deadpond", secret_name="Dive Wilson", team=team_z_force, experience_points=1
         )
         session.add(hero_deadpond)
         session.commit()
diff --git a/docs_src/tutorial/code_structure/tutorial001/models.py b/docs_src/tutorial/code_structure/tutorial001/models.py
@@ -1,8 +1,8 @@
+from pydantic import condecimal
 from typing import List, Optional
 
 from sqlmodel import Field, Relationship, SQLModel
 
-
 class Team(SQLModel, table=True):
     id: Optional[int] = Field(default=None, primary_key=True)
     name: str
@@ -16,6 +16,7 @@ class Hero(SQLModel, table=True):
     name: str
     secret_name: str
     age: Optional[int] = None
+    experience_points: condecimal(max_digits=6, decimal_places=3) = Field(default=0, nullable=False, index=True)
 
     team_id: Optional[int] = Field(default=None, foreign_key="team.id")
     team: Optional[Team] = Relationship(back_populates="heroes")
diff --git a/docs_src/tutorial/code_structure/tutorial002/app.py b/docs_src/tutorial/code_structure/tutorial002/app.py
@@ -10,7 +10,7 @@ def create_heroes():
         team_z_force = Team(name="Z-Force", headquarters="Sister Margaret’s Bar")
 
         hero_deadpond = Hero(
-            name="Deadpond", secret_name="Dive Wilson", team=team_z_force
+            name="Deadpond", secret_name="Dive Wilson", team=team_z_force, experience_points=1
         )
         session.add(hero_deadpond)
         session.commit()
diff --git a/docs_src/tutorial/code_structure/tutorial002/hero_model.py b/docs_src/tutorial/code_structure/tutorial002/hero_model.py
@@ -1,3 +1,4 @@
+from pydantic import condecimal
 from typing import TYPE_CHECKING, Optional
 
 from sqlmodel import Field, Relationship, SQLModel
@@ -11,6 +12,7 @@ class Hero(SQLModel, table=True):
     name: str
     secret_name: str
     age: Optional[int] = None
+    experience_points: condecimal(max_digits=6, decimal_places=3) = Field(default=0, nullable=False, index=True)
 
     team_id: Optional[int] = Field(default=None, foreign_key="team.id")
     team: Optional["Team"] = Relationship(back_populates="heroes")
diff --git a/tests/test_tutorial/test_code_structure/test_tutorial001.py b/tests/test_tutorial/test_code_structure/test_tutorial001.py
@@ -4,6 +4,7 @@
 
 from ...conftest import get_testing_print_function
 
+
 expected_calls = [
     [
         "Created hero:",
@@ -13,6 +14,7 @@
             "age": None,
             "secret_name": "Dive Wilson",
             "team_id": 1,
+            "experience_points": 1,
         },
     ],
     [
diff --git a/tests/test_tutorial/test_code_structure/test_tutorial002.py b/tests/test_tutorial/test_code_structure/test_tutorial002.py
@@ -13,6 +13,7 @@
             "age": None,
             "secret_name": "Dive Wilson",
             "team_id": 1,
+            "experience_points": 1,
         },
     ],
     [

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ def create_heroes():`
`9`	`9`	`team_z_force = Team(name="Z-Force", headquarters="Sister Margaret’s Bar")`
`10`	`10`
`11`	`11`	`hero_deadpond = Hero(`
`12`		`- name="Deadpond", secret_name="Dive Wilson", team=team_z_force`
	`12`	`+ name="Deadpond", secret_name="Dive Wilson", team=team_z_force, experience_points=1`
`13`	`13`	`)`
`14`	`14`	`session.add(hero_deadpond)`
`15`	`15`	`session.commit()`
Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ def create_heroes():`
`10`	`10`	`team_z_force = Team(name="Z-Force", headquarters="Sister Margaret’s Bar")`
`11`	`11`
`12`	`12`	`hero_deadpond = Hero(`
`13`		`- name="Deadpond", secret_name="Dive Wilson", team=team_z_force`
	`13`	`+ name="Deadpond", secret_name="Dive Wilson", team=team_z_force, experience_points=1`
`14`	`14`	`)`
`15`	`15`	`session.add(hero_deadpond)`
`16`	`16`	`session.commit()`
Original file line number	Diff line number	Diff line change
`@@ -4,6 +4,7 @@`
`4`	`4`
`5`	`5`	`from ...conftest import get_testing_print_function`
`6`	`6`
	`7`	`+`
`7`	`8`	`expected_calls = [`
`8`	`9`	`[`
`9`	`10`	`"Created hero:",`
`@@ -13,6 +14,7 @@`
`13`	`14`	`"age": None,`
`14`	`15`	`"secret_name": "Dive Wilson",`
`15`	`16`	`"team_id": 1,`
	`17`	`+ "experience_points": 1,`
`16`	`18`	`},`
`17`	`19`	`],`
`18`	`20`	`[`
Original file line number	Diff line number	Diff line change
`@@ -13,6 +13,7 @@`
`13`	`13`	`"age": None,`
`14`	`14`	`"secret_name": "Dive Wilson",`
`15`	`15`	`"team_id": 1,`
	`16`	`+ "experience_points": 1,`
`16`	`17`	`},`
`17`	`18`	`],`
`18`	`19`	`[`