def setup_test_df():
"""Create a dataframe populated with some test columns."""
df = learn.DataFrame()
df["a"] = learn.TransformedSeries([mocks.MockSeries("foobar", [])],
mocks.MockTwoOutputTransform(
"iue", "eui", "snt"), "out1")
df["b"] = learn.TransformedSeries([mocks.MockSeries("foobar", [])],
mocks.MockTwoOutputTransform(
"iue", "eui", "snt"), "out2")
df["c"] = learn.TransformedSeries([mocks.MockSeries("foobar", [])],
mocks.MockTwoOutputTransform(
"iue", "eui", "snt"), "out1")
return df
def setup_test_df_3layer():
"""Create a dataframe populated with some test columns."""
df = learn.DataFrame()
df["a"] = mocks.MockSeries("a_series", mocks.MockTensor("Tensor a", tf.int32))
df["b"] = mocks.MockSeries("b_series",
mocks.MockSparseTensor("SparseTensor b", tf.int32))
df["c"] = mocks.MockSeries("c_series", mocks.MockTensor("Tensor c", tf.int32))
df["d"] = mocks.MockSeries("d_series",
mocks.MockSparseTensor("SparseTensor d", tf.int32))
df["e"] = learn.TransformedSeries([df["a"], df["b"]],
mocks.Mock2x2Transform("iue", "eui", "snt"),
"out1")
df["f"] = learn.TransformedSeries([df["c"], df["d"]],
mocks.Mock2x2Transform("iue", "eui", "snt"),
"out2")
df["g"] = learn.TransformedSeries([df["e"], df["f"]],
mocks.Mock2x2Transform("iue", "eui", "snt"),
"out1")
return df