/* * MIT License * * Copyright (c) 2019 norangebit * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to deal * in the Software without restriction, including without limitation the rights * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell * copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in all * copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. * */ package it.norangeb.algorithms.graph import it.norangeb.algorithms.graph.data.Edge import org.amshove.kluent.`should be equal to` import org.amshove.kluent.`should equal` import org.spekframework.spek2.Spek import org.spekframework.spek2.style.gherkin.Feature object WeightGraphTest : Spek({ Feature("weight undirected graph") { val graph by memoized { WeightGraph() } Scenario("add edges") { When("add edge 1-4:5.0") { graph.addEdge(Edge(1, 4, 5.0)) } Then("number of edge should equal to 1") { graph.edgeNumber() `should be equal to` 1 } Then("adjacent of 1 should equal listOf(1-4:5.0)") { graph.adjacentVertex(1) `should equal` listOf(Edge(1, 4, 5.0)) } Then("adjacent of 4 should equal listOf(4-1:5.0)") { graph.adjacentVertex(4) `should equal` listOf(Edge(4, 1, 5.0)) } When("add edge 1-3:1.1, 3-4:3.4, 1-5:1.5") { graph.addEdge(Edge(1, 3, 1.1)) graph.addEdge(Edge(3, 4, 3.4)) graph.addEdge(Edge(1, 5, 1.5)) } Then("number of edge should equal to 4") { graph.edgeNumber() `should be equal to` 4 } Then("adjacent of 1 should equal listOf(1-3:1.1, 1-4:5.0, 1-5:1.5)") { graph.adjacentVertex(1).sorted() `should equal` listOf( Edge(1, 3, 1.1), Edge(1, 5, 1.5), Edge(1, 4, 5.0) ) } Then("adjacent of 3 should equal listOf(3-1:1.1, 3-4:3.4)") { graph.adjacentVertex(3).sorted() `should equal` listOf( Edge(3, 1, 1.1), Edge(3, 4, 3.4) ) } Then("edges should equals listOf(1->3:1.3, 1->5:1.5, 3->4:3.4, 1->4:5.0,") { graph.edges().sorted() `should equal` listOf( Edge(1, 3, 1.1), Edge(3, 1, 1.1), Edge(1, 5, 1.5), Edge(5, 1, 1.5), Edge(3, 4, 3.4), Edge(4, 3, 3.4), Edge(1, 4, 5.0), Edge(4, 1, 5.0) ) } } } })