func (s *CollectionFunctorsSuite) TestCollectEdges(c *C) { slice := CollectEdges(spec.GraphLiteralFixture(true)) c.Assert(len(slice), Equals, 2) set1 := set.NewNonTS() for _, e := range slice { set1.Add(e) } c.Assert(set1.Has(NewEdge("foo", "bar")), Equals, true) c.Assert(set1.Has(NewEdge("bar", "baz")), Equals, true) slice2 := CollectEdges(spec.GraphFixtures["2e3v"]) c.Assert(len(slice2), Equals, 2) set2 := set.NewNonTS() for _, e := range slice2 { set2.Add(NewEdge(e.Both())) } c.Assert(set2.Has(NewEdge("foo", "bar")), Equals, true) c.Assert(set2.Has(NewEdge("bar", "baz")), Equals, true) }
func (s *CollectionFunctorsSuite) TestCollectVertices(c *C) { slice := CollectVertices(spec.GraphLiteralFixture(true)) c.Assert(len(slice), Equals, 4) set1 := set.NewNonTS() for _, v := range slice { set1.Add(v) } c.Assert(set1.Has("foo"), Equals, true) c.Assert(set1.Has("bar"), Equals, true) c.Assert(set1.Has("baz"), Equals, true) c.Assert(set1.Has("isolate"), Equals, true) slice2 := CollectVertices(spec.GraphFixtures["2e3v"]) c.Assert(len(slice2), Equals, 3) set2 := set.NewNonTS() for _, v := range slice2 { set2.Add(v) } c.Assert(set2.Has("foo"), Equals, true) c.Assert(set2.Has("bar"), Equals, true) c.Assert(set2.Has("baz"), Equals, true) }
func (s *CountingFunctorsSuite) TestSize(c *C) { el := EdgeList{ NewEdge("foo", "bar"), NewEdge("bar", "baz"), NewEdge("foo", "qux"), NewEdge("qux", "bar"), } c.Assert(Size(el), Equals, 4) c.Assert(Size(spec.GraphLiteralFixture(true)), Equals, 2) }
func (s *CollectionFunctorsSuite) TestCollectEdgesIncidentTo(c *C) { slice := CollectEdgesIncidentTo("foo", spec.GraphLiteralFixture(true)) c.Assert(len(slice), Equals, 1) set := set.NewNonTS() for _, e := range slice { set.Add(e) } c.Assert(set.Has(NewEdge("foo", "bar")), Equals, true) }
func (s *CollectionFunctorsSuite) TestCollectAdjacentVertices(c *C) { slice := CollectVerticesAdjacentTo("bar", spec.GraphLiteralFixture(true)) c.Assert(len(slice), Equals, 2) set := set.NewNonTS() for _, v := range slice { set.Add(v) } c.Assert(set.Has("foo"), Equals, true) c.Assert(set.Has("baz"), Equals, true) }