Define schema_alter in test module

Per @A-Baji suggestion

Author: ethho

tests/schema_alter.py

@@ -1,12 +1,66 @@
 import pytest
 import re
 import datajoint as dj
-from . import schema as schema_any_module, schema_alter as schema_alter_module, PREFIX
-from .schema_alter import Parent, Experiment
+from . import schema as schema_any_module, PREFIX
 
+
+class Experiment(dj.Imported):
+    original_definition = """  # information about experiments
+    -> Subject
+    experiment_id  :smallint  # experiment number for this subject
+    ---
+    experiment_date  :date   # date when experiment was started
+    -> [nullable] User
+    data_path=""     :varchar(255)  # file path to recorded data
+    notes=""         :varchar(2048) # e.g. purpose of experiment
+    entry_time=CURRENT_TIMESTAMP :timestamp   # automatic timestamp
+    """
+
+    definition1 = """  # Experiment
+    -> Subject
+    experiment_id  :smallint  # experiment number for this subject
+    ---
+    data_path     : int  # some number
+    extra=null : longblob  # just testing
+    -> [nullable] User
+    subject_notes=null         :varchar(2048) # {notes} e.g. purpose of experiment
+    entry_time=CURRENT_TIMESTAMP :timestamp   # automatic timestamp
+    """
+
+
+class Parent(dj.Manual):
+    definition = """
+    parent_id: int
+    """
+
+    class Child(dj.Part):
+        definition = """
+        -> Parent
+        """
+        definition_new = """
+        -> master
+        ---
+        child_id=null: int
+        """
+
+    class Grandchild(dj.Part):
+        definition = """
+        -> master.Child
+        """
+        definition_new = """
+        -> master.Child
+        ---
+        grandchild_id=null: int
+        """
+
+
+LOCALS_ALTER = {
+    "Experiment": Experiment,
+    "Parent": Parent
+}
 COMBINED_CONTEXT = {
     **schema_any_module.LOCALS_ANY,
-    **schema_alter_module.LOCALS_ALTER,
+    **LOCALS_ALTER,
 }
 
 
@@ -54,9 +108,9 @@ def schema_alter(connection_test):
     schema_any(schema_any_module.Stimulus)
     schema_any(schema_any_module.Longblob)
 
-    # Add nodes from schema_alter_module
-    schema_any(Experiment, context=schema_alter_module.LOCALS_ALTER)
-    schema_any(Parent, context=schema_alter_module.LOCALS_ALTER)
+    # Overwrite Experiment and Parent nodes
+    schema_any(Experiment, context=LOCALS_ALTER)
+    schema_any(Parent, context=LOCALS_ALTER)
 
     yield schema_any
     schema_any.drop()