added test

rleap-project · Mar 6, 2024 · 9caf93e · 9caf93e
1 parent d6e993b
commit 9caf93e
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/tests/novelty/gripper/gripper.cpp b/tests/novelty/gripper/gripper.cpp
@@ -15,7 +15,7 @@ namespace dlplan::tests::novelty {
 
 TEST(DLPTests, NoveltyGripperTest) {
     auto result = generate_state_space("domain.pddl", "p-1-0.pddl");
-    std::shared_ptr<StateSpace> state_space = std::move(result.state_space);
+    auto state_space = std::move(result.state_space);
 
     auto novelty_base_0 = std::make_shared<NoveltyBase>(
         state_space->get_instance_info()->get_atoms().size(),
@@ -30,18 +30,20 @@ TEST(DLPTests, NoveltyGripperTest) {
     EXPECT_EQ(TupleGraph(novelty_base_1, state_space, 0, false).compute_repr(), "TupleGraph(\n  root_state_index=0,\n  tuple_nodes=[    TupleNode(index=0, tuple_index=0, state_indices={0}, predecessors=[], successors=[2, 6, 7]),\n    TupleNode(index=1, tuple_index=1, state_indices={0}, predecessors=[], successors=[2, 6, 7]),\n    TupleNode(index=2, tuple_index=2, state_indices={1}, predecessors=[0, 1, 3, 4, 5], successors=[]),\n    TupleNode(index=3, tuple_index=3, state_indices={0}, predecessors=[], successors=[2, 6, 7]),\n    TupleNode(index=4, tuple_index=4, state_indices={0}, predecessors=[], successors=[2, 6, 7]),\n    TupleNode(index=5, tuple_index=5, state_indices={0}, predecessors=[], successors=[2, 6, 7]),\n    TupleNode(index=6, tuple_index=7, state_indices={2}, predecessors=[0, 1, 3, 4, 5], successors=[]),\n    TupleNode(index=7, tuple_index=8, state_indices={3}, predecessors=[0, 1, 3, 4, 5], successors=[])\n  ],\n  node_indices_by_distance=[\n    [0, 1, 3, 4, 5],\n    [2, 6, 7]\n  ],\n  state_indices_by_distance=[\n    [0],\n    [1, 2, 3]\n  ]\n)");
     EXPECT_EQ(TupleGraph(novelty_base_1, state_space, 0, true).compute_repr(), "TupleGraph(\n  root_state_index=0,\n  tuple_nodes=[    TupleNode(index=0, tuple_index=0, state_indices={0}, predecessors=[], successors=[1, 2, 3]),\n    TupleNode(index=1, tuple_index=2, state_indices={1}, predecessors=[0], successors=[]),\n    TupleNode(index=2, tuple_index=7, state_indices={2}, predecessors=[0], successors=[]),\n    TupleNode(index=3, tuple_index=8, state_indices={3}, predecessors=[0], successors=[])\n  ],\n  node_indices_by_distance=[\n    [0],\n    [1, 2, 3]\n  ],\n  state_indices_by_distance=[\n    [0],\n    [1, 2, 3]\n  ]\n)");
 
-    std::cout << std::endl;
     auto result2 = generate_state_space("domain.pddl", "p-2-0.pddl");
     auto state_space2 = result2.state_space;
     auto novelty_base_2 = std::make_shared<NoveltyBase>(
         state_space2->get_instance_info()->get_atoms().size(),
-        1);
-    std::ofstream file("out.dot");
-    const auto tg = TupleGraph(novelty_base_2, state_space2, 2, true);
-    file << tg.to_dot(1);
-    file.close();
+        2);
+
+    EXPECT_EQ(TupleGraph(novelty_base_2, state_space2, 0, true).compute_repr(), "TupleGraph(\n  root_state_index=0,\n  tuple_nodes=[    TupleNode(index=0, tuple_index=0, state_indices={0}, predecessors=[], successors=[1, 5, 8, 12, 15]),\n    TupleNode(index=1, tuple_index=54, state_indices={1}, predecessors=[0], successors=[]),\n    TupleNode(index=2, tuple_index=79, state_indices={16}, predecessors=[3, 7], successors=[]),\n    TupleNode(index=3, tuple_index=82, state_indices={12}, predecessors=[6], successors=[2]),\n    TupleNode(index=4, tuple_index=93, state_indices={6}, predecessors=[8], successors=[7]),\n    TupleNode(index=5, tuple_index=104, state_indices={4}, predecessors=[0], successors=[6, 14]),\n    TupleNode(index=6, tuple_index=106, state_indices={10}, predecessors=[5], successors=[3, 7]),\n    TupleNode(index=7, tuple_index=123, state_indices={12}, predecessors=[4, 6], successors=[2]),\n    TupleNode(index=8, tuple_index=124, state_indices={2}, predecessors=[0], successors=[4, 17]),\n    TupleNode(index=9, tuple_index=131, state_indices={19}, predecessors=[10, 11], successors=[]),\n    TupleNode(index=10, tuple_index=134, state_indices={14}, predecessors=[13], successors=[9]),\n    TupleNode(index=11, tuple_index=135, state_indices={14}, predecessors=[13, 16], successors=[9]),\n    TupleNode(index=12, tuple_index=143, state_indices={3}, predecessors=[0], successors=[13, 14]),\n    TupleNode(index=13, tuple_index=145, state_indices={8}, predecessors=[12], successors=[10, 11]),\n    TupleNode(index=14, tuple_index=151, state_indices={9}, predecessors=[5, 12], successors=[]),\n    TupleNode(index=15, tuple_index=156, state_indices={5}, predecessors=[0], successors=[16, 17]),\n    TupleNode(index=16, tuple_index=158, state_indices={11}, predecessors=[15], successors=[11]),\n    TupleNode(index=17, tuple_index=163, state_indices={7}, predecessors=[8, 15], successors=[])\n  ],\n  node_indices_by_distance=[\n    [0],\n    [1, 5, 8, 12, 15],\n    [4, 6, 13, 14, 16, 17],\n    [3, 7, 10, 11],\n    [2, 9]\n  ],\n  state_indices_by_distance=[\n    [0],\n    [1, 2, 3, 4, 5],\n    [6, 7, 8, 9, 10, 11],\n    [12, 13, 14, 15],\n    [16, 17, 18, 19, 20, 21]\n  ]\n)");
+
+    //std::ofstream file("out.dot");
+    //const auto tg = TupleGraph(novelty_base_2, state_space2, 2, true);
+    //file << tg.to_dot(1);
+    //file.close();
+    //std::cout << tg.compute_repr() << std::endl;
 
-    std::cout << tg.compute_repr() << std::endl;
 }
 
 }