add tests

This commit is contained in:
evilchili
2024-08-20 09:30:23 -07:00
parent b7732f1581
commit 9f75630c74
4 changed files with 111 additions and 11 deletions
+1 -1
View File
@@ -328,7 +328,7 @@ class Character(BaseObject, SlugMixin, ModifierMixin):
merge_modifiers(self.conditions)
for mapping in self.equipped_items:
for (target, mods) in mapping.item.modifiers.items():
for target, mods in mapping.item.modifiers.items():
for mod in mods:
if mod.requires_attunement and not mapping.attuned:
continue
+2 -9
View File
@@ -2,9 +2,9 @@ from sqlalchemy import ForeignKey, String
from sqlalchemy.orm import Mapped, mapped_column, relationship
from ttfrog.db.base import BaseObject, EnumField
from ttfrog.db.schema.classes import CharacterClass
from ttfrog.db.schema.constants import DamageType
from ttfrog.db.schema.modifiers import ModifierMixin
from ttfrog.db.schema.classes import CharacterClass
__all__ = [
"Item",
@@ -21,14 +21,7 @@ ITEM_TYPES = [
"SHIELD",
]
RARITY = [
"Common",
"Uncommon",
"Rare",
"Very Rare",
"Legendary",
"Artifact"
]
RARITY = ["Common", "Uncommon", "Rare", "Very Rare", "Legendary", "Artifact"]
ItemType = EnumField("ItemType", ((k, k) for k in ITEM_TYPES))
+1 -1
View File
@@ -175,7 +175,7 @@ class ModifierMixin:
Returns the matching column if it was found, or None.
"""
if attr_name.startswith('_'):
if attr_name.startswith("_"):
return None
col = getattr(self.__table__.columns, f"_{attr_name}", None)
if col is None: