Fix node conversion tests

ryukzak · Aug 31, 2023 · 0369b28 · 0369b28
1 parent ffa27ff
commit 0369b28
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/ml/synthesis/src/tests/fixtures/test_tree_root.py b/ml/synthesis/src/tests/fixtures/test_tree_root.py
@@ -20,7 +20,7 @@ def test_root():
                 is_terminal=True,
                 is_finish=True,
                 parameters={"a": 1, "b": 2},
-                decision=dict(tag="BindDecisionView"),
+                decision=dict(tag="SingleBindView"),
                 duration=None,
                 sid="-0",
                 children=[],

diff --git a/ml/synthesis/src/tests/test_node_json_conversion.py b/ml/synthesis/src/tests/test_node_json_conversion.py
@@ -13,9 +13,7 @@ def test_node_from_raw_dict(test_root, test_root_dict):
 def test_node_to_train_df_dict(test_root: NittaNodeInTree):
     assert test_root.children, "test root must have children for this test to run"
     assert nitta_node_to_df_dict(test_root, siblings=test_root.children, example="test") == {
-        "alt_bindings": 1,
-        "alt_dataflows": 0,
-        "alt_refactorings": 0,
+        "siblings_SingleBindView": 1,
         "example": "test",
         "is_terminal": False,
         "old_score": 1010,