7630174820
* update relation * feat: add relation_type to avatar info structure and update related components - Added `relation_type` to the avatar structured info in `info_presenter.py`. - Updated `AvatarDetail.vue` to utilize the new `relation_type` for displaying avatar relationships. - Modified `RelationRow.vue` to accept `type` as a prop for enhanced relationship representation. - Updated `core.ts` to include `relation_type` in the `RelationInfo` interface. Closes #
92 lines
3.6 KiB
Python
92 lines
3.6 KiB
Python
import pytest
|
|
from src.classes.relation.relation import Relation
|
|
from src.classes.relation.relations import update_second_degree_relations, set_relation
|
|
from src.classes.avatar import Avatar, Gender
|
|
from src.classes.age import Age
|
|
from src.classes.cultivation import Realm
|
|
from src.classes.calendar import MonthStamp
|
|
from src.utils.id_generator import get_avatar_id
|
|
|
|
def create_avatar(world, name, gender=Gender.MALE):
|
|
return Avatar(
|
|
world=world,
|
|
name=name,
|
|
id=get_avatar_id(),
|
|
birth_month_stamp=MonthStamp(0),
|
|
age=Age(20, Realm.Qi_Refinement),
|
|
gender=gender,
|
|
pos_x=0, pos_y=0
|
|
)
|
|
|
|
def test_family_relations(base_world):
|
|
grandpa = create_avatar(base_world, "Grandpa")
|
|
father = create_avatar(base_world, "Father")
|
|
son = create_avatar(base_world, "Son")
|
|
daughter = create_avatar(base_world, "Daughter", Gender.FEMALE)
|
|
|
|
# Setup relations: Grandpa -> Father -> Son/Daughter
|
|
# Father's parent is Grandpa
|
|
set_relation(father, grandpa, Relation.PARENT)
|
|
# Son's parent is Father
|
|
set_relation(son, father, Relation.PARENT)
|
|
# Daughter's parent is Father
|
|
set_relation(daughter, father, Relation.PARENT)
|
|
|
|
# Update logic
|
|
for p in [grandpa, father, son, daughter]:
|
|
update_second_degree_relations(p)
|
|
|
|
# Assertions
|
|
|
|
# 1. Sibling check (Son <-> Daughter)
|
|
# Son perspective
|
|
assert son.computed_relations.get(daughter) == Relation.SIBLING
|
|
# Daughter perspective
|
|
assert daughter.computed_relations.get(son) == Relation.SIBLING
|
|
|
|
# 2. Grandparent check (Son/Daughter -> Grandpa)
|
|
assert son.computed_relations.get(grandpa) == Relation.GRAND_PARENT
|
|
assert daughter.computed_relations.get(grandpa) == Relation.GRAND_PARENT
|
|
|
|
# 3. Grandchild check (Grandpa -> Son/Daughter)
|
|
assert grandpa.computed_relations.get(son) == Relation.GRAND_CHILD
|
|
assert grandpa.computed_relations.get(daughter) == Relation.GRAND_CHILD
|
|
|
|
# 4. Father should not have Sibling/Grandparent (in this limited set)
|
|
assert Relation.SIBLING not in father.computed_relations.values()
|
|
assert Relation.GRAND_PARENT not in father.computed_relations.values()
|
|
|
|
def test_sect_relations(base_world):
|
|
master = create_avatar(base_world, "Master")
|
|
disciple_a = create_avatar(base_world, "DiscipleA")
|
|
disciple_b = create_avatar(base_world, "DiscipleB")
|
|
grand_master = create_avatar(base_world, "GrandMaster")
|
|
|
|
# Setup: GrandMaster -> Master -> A/B
|
|
# Master is disciple of GrandMaster
|
|
set_relation(master, grand_master, Relation.MASTER) # master.relations[GM] = MASTER, GM.relations[master] = APPRENTICE
|
|
|
|
# A is disciple of Master
|
|
set_relation(disciple_a, master, Relation.MASTER)
|
|
|
|
# B is disciple of Master
|
|
set_relation(disciple_b, master, Relation.MASTER)
|
|
|
|
# Update
|
|
for p in [grand_master, master, disciple_a, disciple_b]:
|
|
update_second_degree_relations(p)
|
|
|
|
# Assertions
|
|
|
|
# 1. Martial Sibling (A <-> B)
|
|
assert disciple_a.computed_relations.get(disciple_b) == Relation.MARTIAL_SIBLING
|
|
assert disciple_b.computed_relations.get(disciple_a) == Relation.MARTIAL_SIBLING
|
|
|
|
# 2. Martial Grandmaster (A/B -> GrandMaster)
|
|
assert disciple_a.computed_relations.get(grand_master) == Relation.MARTIAL_GRANDMASTER
|
|
assert disciple_b.computed_relations.get(grand_master) == Relation.MARTIAL_GRANDMASTER
|
|
|
|
# 3. Martial Grandchild (GrandMaster -> A/B)
|
|
assert grand_master.computed_relations.get(disciple_a) == Relation.MARTIAL_GRANDCHILD
|
|
assert grand_master.computed_relations.get(disciple_b) == Relation.MARTIAL_GRANDCHILD
|