Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: rename "example_config" -> "example_it_config"

Browse Source
This commit is contained in:
psychedelicious 2024-03-05 12:05:05 +11:00
parent 5551cf8ac4
commit 6829610a71
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/app/services/model_records/test_model_records_sql.py
View File

@ -40,7 +40,7 @@ def store(
return ModelRecordServiceSQL(db) return ModelRecordServiceSQL(db)
def example_config(key: Optional[str] = None) -> TextualInversionFileConfig: def example_ti_config(key: Optional[str] = None) -> TextualInversionFileConfig:
config = TextualInversionFileConfig( config = TextualInversionFileConfig(
source="test/source/", source="test/source/",
source_type=ModelSourceType.Path, source_type=ModelSourceType.Path,
@ -57,7 +57,7 @@ def example_config(key: Optional[str] = None) -> TextualInversionFileConfig:
def test_type(store: ModelRecordServiceBase): def test_type(store: ModelRecordServiceBase):
config = example_config("key1") config = example_ti_config("key1")
store.add_model(config) store.add_model(config)
config1 = store.get_model("key1") config1 = store.get_model("key1")
assert isinstance(config1, TextualInversionFileConfig) assert isinstance(config1, TextualInversionFileConfig)
@ -65,7 +65,7 @@ def test_type(store: ModelRecordServiceBase):
def test_raises_on_violating_uniqueness(store: ModelRecordServiceBase): def test_raises_on_violating_uniqueness(store: ModelRecordServiceBase):
# Models have a uniqueness constraint by their name, base and type # Models have a uniqueness constraint by their name, base and type
config1 = example_config("key1") config1 = example_ti_config("key1")
config2 = config1.model_copy(deep=True) config2 = config1.model_copy(deep=True)
config2.key = "key2" config2.key = "key2"
store.add_model(config1) store.add_model(config1)
@ -76,7 +76,7 @@ def test_raises_on_violating_uniqueness(store: ModelRecordServiceBase):
def test_model_records_updates_model(store: ModelRecordServiceBase): def test_model_records_updates_model(store: ModelRecordServiceBase):
config = example_config("key1") config = example_ti_config("key1")
store.add_model(config) store.add_model(config)
config = store.get_model("key1") config = store.get_model("key1")
assert config.name == "old name" assert config.name == "old name"
@ -88,7 +88,7 @@ def test_model_records_updates_model(store: ModelRecordServiceBase):
def test_model_records_rejects_invalid_changes(store: ModelRecordServiceBase): def test_model_records_rejects_invalid_changes(store: ModelRecordServiceBase):
config = example_config("key1") config = example_ti_config("key1")
store.add_model(config) store.add_model(config)
config = store.get_model("key1") config = store.get_model("key1")
# upcast_attention is an invalid field for TIs # upcast_attention is an invalid field for TIs
@ -98,14 +98,14 @@ def test_model_records_rejects_invalid_changes(store: ModelRecordServiceBase):
def test_unknown_key(store: ModelRecordServiceBase): def test_unknown_key(store: ModelRecordServiceBase):
config = example_config("key1") config = example_ti_config("key1")
store.add_model(config) store.add_model(config)
with pytest.raises(UnknownModelException): with pytest.raises(UnknownModelException):
store.update_model("unknown_key", ModelRecordChanges()) store.update_model("unknown_key", ModelRecordChanges())
def test_delete(store: ModelRecordServiceBase): def test_delete(store: ModelRecordServiceBase):
config = example_config("key1") config = example_ti_config("key1")
store.add_model(config) store.add_model(config)
config = store.get_model("key1") config = store.get_model("key1")
store.del_model("key1") store.del_model("key1")
@ -114,7 +114,7 @@ def test_delete(store: ModelRecordServiceBase):
def test_exists(store: ModelRecordServiceBase): def test_exists(store: ModelRecordServiceBase):
config = example_config("key1") config = example_ti_config("key1")
store.add_model(config) store.add_model(config)
assert store.exists("key1") assert store.exists("key1")
assert not store.exists("key2") assert not store.exists("key2")