Skip to content

Commit 8a8b0c3

Browse files
committed
Simpler schema_alter fixture
Per @A-Baji suggestion
1 parent e2c0d30 commit 8a8b0c3

File tree

1 file changed

+2
-48
lines changed

1 file changed

+2
-48
lines changed

tests/test_alter.py

Lines changed: 2 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -54,64 +54,18 @@ class Grandchild(dj.Part):
5454
"""
5555

5656

57-
LOCALS_ALTER = {
58-
"Experiment": Experiment,
59-
"Parent": Parent
60-
}
57+
LOCALS_ALTER = {"Experiment": Experiment, "Parent": Parent}
6158
COMBINED_CONTEXT = {
6259
**schema_any_module.LOCALS_ANY,
6360
**LOCALS_ALTER,
6461
}
6562

6663

6764
@pytest.fixture
68-
def schema_alter(connection_test):
69-
schema_any = dj.Schema(
70-
PREFIX + "_alter",
71-
context=schema_any_module.LOCALS_ANY,
72-
connection=connection_test,
73-
)
74-
schema_any(schema_any_module.TTest)
75-
schema_any(schema_any_module.TTest2)
76-
schema_any(schema_any_module.TTest3)
77-
schema_any(schema_any_module.NullableNumbers)
78-
schema_any(schema_any_module.TTestExtra)
79-
schema_any(schema_any_module.TTestNoExtra)
80-
schema_any(schema_any_module.Auto)
81-
schema_any(schema_any_module.User)
82-
schema_any(schema_any_module.Subject)
83-
schema_any(schema_any_module.Language)
84-
schema_any(schema_any_module.Experiment)
85-
schema_any(schema_any_module.Trial)
86-
schema_any(schema_any_module.Ephys)
87-
schema_any(schema_any_module.Image)
88-
schema_any(schema_any_module.UberTrash)
89-
schema_any(schema_any_module.UnterTrash)
90-
schema_any(schema_any_module.SimpleSource)
91-
schema_any(schema_any_module.SigIntTable)
92-
schema_any(schema_any_module.SigTermTable)
93-
schema_any(schema_any_module.DjExceptionName)
94-
schema_any(schema_any_module.ErrorClass)
95-
schema_any(schema_any_module.DecimalPrimaryKey)
96-
schema_any(schema_any_module.IndexRich)
97-
schema_any(schema_any_module.ThingA)
98-
schema_any(schema_any_module.ThingB)
99-
schema_any(schema_any_module.ThingC)
100-
schema_any(schema_any_module.Parent)
101-
schema_any(schema_any_module.Child)
102-
schema_any(schema_any_module.ComplexParent)
103-
schema_any(schema_any_module.ComplexChild)
104-
schema_any(schema_any_module.SubjectA)
105-
schema_any(schema_any_module.SessionA)
106-
schema_any(schema_any_module.SessionStatusA)
107-
schema_any(schema_any_module.SessionDateA)
108-
schema_any(schema_any_module.Stimulus)
109-
schema_any(schema_any_module.Longblob)
110-
65+
def schema_alter(connection_test, schema_any):
11166
# Overwrite Experiment and Parent nodes
11267
schema_any(Experiment, context=LOCALS_ALTER)
11368
schema_any(Parent, context=LOCALS_ALTER)
114-
11569
yield schema_any
11670
schema_any.drop()
11771

0 commit comments

Comments
 (0)