code: def test_feature_trie_with_needs_full_dataframe(diamond_es):
      pd_es = diamond_es
      amount = IdentityFeature(pd_es["transactions"].ww["amount"])
  
      path_through_customers = backward_path(
          pd_es,
          ["regions", "customers", "transactions"],
      )
      agg = AggregationFeature(
          amount,
          "regions",
          primitive=primitives.Mean,
          relationship_path=path_through_customers,
      )
      trans_of_agg = TransformFeature(agg, primitives.CumSum)
  
      path_through_stores = backward_path(
          pd_es, ["regions", "stores", "transactions"]
      )
      trans = TransformFeature(amount, primitives.CumSum)
      agg_of_trans = AggregationFeature(
          trans,
          "regions",
          primitive=primitives.Mean,
          relationship_path=path_through_stores,
      )
  
      features = [agg, trans_of_agg, agg_of_trans]
      feature_set = FeatureSet(features)
      trie = feature_set.feature_trie
  
      assert trie.value == (
          True,
          {agg.unique_name(), trans_of_agg.unique_name()},
          {agg_of_trans.unique_name()},
      )
      assert trie.get_node(path_through_customers).value == (
          True,
          {amount.unique_name()},
          set(),
      )
      assert trie.get_node(path_through_customers[:1]).value == (
          True,
          set(),
          set(),
      )
      assert trie.get_node(path_through_stores).value == (
          True,
          {amount.unique_name(), trans.unique_name()},
          set(),
      )
      assert trie.get_node(path_through_stores[:1]).value == (
          False,
          set(),
          set(),
      )
  
file_path: featuretools/featuretools/tests/computational_backend/test_feature_set.py
imports: from featuretools import AggregationFeature, DirectFeature, IdentityFeature, TransformFeature, primitives
  from featuretools.computational_backends.feature_set import FeatureSet
  from featuretools.entityset.relationship import RelationshipPath
  from featuretools.tests.testing_utils import backward_path
  from featuretools.utils import Trie
label_desc: Use image node_id 2 for calling a global function with example usage: test_feature_trie_with_needs_full_dataframe(diamond_es) without return types