add test for GATContext -> Schema

src/categorical_algebra/ACSetsGATsInterop.jl

@@ -28,13 +28,15 @@ end
 function Schema(c::GATContext)
   obs, attrtypes = Symbol[], Symbol[]
   homs, attrs = Tuple{Symbol, Symbol, Symbol}[], Tuple{Symbol, Symbol, Symbol}[]
-  for binding in p.scope
-    type = getvalue(binding)
-    @match (nameof(type.body.head), type.body.args...) begin
-      (:Ob,) => push!(obs, nameof(binding))
-      (:Hom, x, y) => push!(homs, nameof.((binding, x.body, y.body)))
-      (:AttrType,) => push!(attrtypes, nameof(binding))
-      (:Attr, x, y) => push!(attrs, nameof.((binding, x.body, y.body)))
+  for scope in allscopes(gettypecontext(c))
+    for binding in scope
+      type = getvalue(binding)
+      @match (nameof(type.body.head), type.body.args...) begin
+        (:Ob,) => push!(obs, nameof(binding))
+        (:Hom, x, y) => push!(homs, nameof.((binding, x.body, y.body)))
+        (:AttrType,) => push!(attrtypes, nameof(binding))
+        (:Attr, x, y) => push!(attrs, nameof.((binding, x.body, y.body)))
+      end
     end
   end
   BasicSchema(obs,homs,attrtypes,attrs)

test/categorical_algebra/ACSetsGATsInterop.jl

@@ -1,6 +1,7 @@
 module TestACSetsGATsInterop
 using Test
 
+using GATlab
 using Catlab.CategoricalAlgebra
 using Catlab.Graphs: SchGraph, SchWeightedGraph
 
@@ -29,6 +30,15 @@ X, parent, height = SchDendrogram[[:X, :parent, :height]]
 @test subpart(d, id(X)) == 1:5
 @test subpart(d, compose(parent, height)) == [10,10,10,20,20]
 
+@gatcontext SchDendrogram′(ThSchema) begin
+  X::Ob
+  R::AttrType
+  parent::Hom(X,X)
+  height::Attr(X,R)
+end
+
+@test Schema(SchDendrogram′) == Schema(SchDendrogram)
+
 # JSON serialization
 #-------------------
 
@@ -44,4 +54,5 @@ for schema in [SchGraph, SchWeightedGraph]
   @test roundtrip_json_acset_schema(schema) == schema
 end
 
+
 end