 utils = require "../../utils"
 
 {StaticLayoutProvider} = utils.require("models/graphs/static_layout_provider")
-{GraphDataSource} = utils.require("models/graphs/graph_data_source")
+{GraphSource} = utils.require("models/graphs/graph_source")
 
 describe "StaticLayoutProvider", ->
 

    describe "get_node_coordinates method", ->

      it "should return node coords if exist", ->
        graph_source = new GraphDataSource()
        graph_source.nodes.data.index = [0,1,2,3]

        [xs, ys] = @layout_provider.get_node_coordinates(graph_source)

        expect(ys).to.be.deep.equal([ [ 0, 1 ], [ 0, 0 ], [ 0, -1 ] ])

      it "should return nulls if coords don't exist", ->
        graph_source = new GraphSource()
        graph_source.nodes.data.index = [4,5,6,7]
        graph_source.edges.data.start = [4,4,4]
        graph_source.edges.data.end = [5,6,7]
