fix: expand api (make similar to js implementation)

Makefile

@@ -6,6 +6,7 @@ vet:
 	go vet github.com/piprate/json-gold/...
 
 test: vet
+	go clean -testcache
 	go test github.com/piprate/json-gold/...
 
 test-cov: vet

ld/api_expand.go

@@ -790,10 +790,11 @@ func (api *JsonLdApi) expandObject(activeCtx *Context, activeProperty string, ex
 			}
 		}
 
+		isContainer := termCtx.HasContainerMapping2(td, "@graph")
 		isContainerGraph := termCtx.HasContainerMapping(key, "@graph")
 		isContainerID := termCtx.HasContainerMapping(key, "@id")
 		isContainerIndex := termCtx.HasContainerMapping(key, "@index")
-		if isContainerGraph && !isContainerID && !isContainerIndex {
+		if (isContainerGraph || isContainer) && !isContainerID && !isContainerIndex {
 			evList := Arrayify(expandedValue)
 			rVal := make([]interface{}, 0)
 			for _, ev := range evList {

ld/context.go

@@ -1720,6 +1720,18 @@ func (c *Context) HasContainerMapping(property string, val string) bool {
 	return false
 }
 
+func (c *Context) HasContainerMapping2(dict map[string]interface{}, val string) bool {
+	if container, hasContainer := dict["@container"]; hasContainer {
+		for _, container := range container.([]interface{}) {
+			if container == val {
+				return true
+			}
+		}
+	}
+
+	return false
+}
+
 // IsReverseProperty returns true if the given property is a reverse property
 func (c *Context) IsReverseProperty(property string) bool {
 	td := c.GetTermDefinition(property)