A test for complex task logic
This commit is contained in:
parent
74b5a1ac50
commit
a0abb99d88
@ -1,11 +1,11 @@
|
||||
package hep.dataforge.workspace
|
||||
|
||||
import hep.dataforge.context.PluginTag
|
||||
import hep.dataforge.data.first
|
||||
import hep.dataforge.data.get
|
||||
import hep.dataforge.data.*
|
||||
import hep.dataforge.meta.boolean
|
||||
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.assertTrue
|
||||
|
||||
@ -33,7 +33,7 @@ class SimpleWorkspaceTest {
|
||||
}
|
||||
|
||||
|
||||
task("square") {
|
||||
val square = task("square") {
|
||||
model {
|
||||
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") {
|
||||
model {
|
||||
dependsOn("square")
|
||||
@ -107,4 +138,10 @@ class SimpleWorkspaceTest {
|
||||
assertTrue { tasks["test.test"] != null }
|
||||
//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