A test for complex task logic
This commit is contained in:
parent
74b5a1ac50
commit
a0abb99d88
@ -1,11 +1,11 @@
|
|||||||
package hep.dataforge.workspace
|
package hep.dataforge.workspace
|
||||||
|
|
||||||
import hep.dataforge.context.PluginTag
|
import hep.dataforge.context.PluginTag
|
||||||
import hep.dataforge.data.first
|
import hep.dataforge.data.*
|
||||||
import hep.dataforge.data.get
|
|
||||||
import hep.dataforge.meta.boolean
|
import hep.dataforge.meta.boolean
|
||||||
import hep.dataforge.meta.get
|
import hep.dataforge.meta.get
|
||||||
import org.junit.Test
|
import hep.dataforge.names.asName
|
||||||
|
import kotlin.test.Test
|
||||||
import kotlin.test.assertEquals
|
import kotlin.test.assertEquals
|
||||||
import kotlin.test.assertTrue
|
import kotlin.test.assertTrue
|
||||||
|
|
||||||
@ -33,7 +33,7 @@ class SimpleWorkspaceTest {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
task("square") {
|
val square = task("square") {
|
||||||
model {
|
model {
|
||||||
allData()
|
allData()
|
||||||
}
|
}
|
||||||
@ -46,6 +46,37 @@ class SimpleWorkspaceTest {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
val linear = task("linear") {
|
||||||
|
model {
|
||||||
|
allData()
|
||||||
|
}
|
||||||
|
pipe<Int, Int> { data ->
|
||||||
|
context.logger.info { "Starting linear on $data" }
|
||||||
|
data * 2 + 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
val fullSquare = task("fullsquare") {
|
||||||
|
model {
|
||||||
|
dependsOn("square", placement = "square".asName())
|
||||||
|
dependsOn("linear", placement = "linear".asName())
|
||||||
|
}
|
||||||
|
transform<Any> { data ->
|
||||||
|
val squareNode = data["square"].withType<Int>().node!!
|
||||||
|
val linearNode = data["linear"].withType<Int>().node!!
|
||||||
|
return@transform DataNode.build(Int::class) {
|
||||||
|
squareNode.dataSequence().forEach { (name, _) ->
|
||||||
|
val newData = Data{
|
||||||
|
val squareValue = squareNode[name].data!!.get()
|
||||||
|
val linearValue = linearNode[name].data!!.get()
|
||||||
|
squareValue+linearValue
|
||||||
|
}
|
||||||
|
set(name,newData)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
task("sum") {
|
task("sum") {
|
||||||
model {
|
model {
|
||||||
dependsOn("square")
|
dependsOn("square")
|
||||||
@ -107,4 +138,10 @@ class SimpleWorkspaceTest {
|
|||||||
assertTrue { tasks["test.test"] != null }
|
assertTrue { tasks["test.test"] != null }
|
||||||
//val node = workspace.run("test.test", "empty")
|
//val node = workspace.run("test.test", "empty")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun testFullSquare(){
|
||||||
|
val node = workspace.run("fullsquare")
|
||||||
|
println(node.toMeta())
|
||||||
|
}
|
||||||
}
|
}
|
Loading…
Reference in New Issue
Block a user