From 5b11605e2d24f9c42471c41577da10682a7d45e6 Mon Sep 17 00:00:00 2001
From: Arman Aydin <86849247+goldenJester@users.noreply.github.com>
Date: Tue, 20 Feb 2024 14:03:08 -0800
Subject: [PATCH] Readd slicing tests

---
 tests/test_transition_rules.py | 77 ++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)

diff --git a/tests/test_transition_rules.py b/tests/test_transition_rules.py
index 22c4f1156..20101d772 100644
--- a/tests/test_transition_rules.py
+++ b/tests/test_transition_rules.py
@@ -1546,3 +1546,80 @@ def test_single_toggleable_machine_rule_output_object_success():
         obj = DatasetObject(**obj_cfg)
         og.sim.import_object(obj)
     og.sim.step()
+
+@og_test
+def test_slicer_volume_sharp_end():
+    """
+    Demo of slicing an apple into two apple slices
+    """
+
+    # Create the scene config to load -- empty scene with knife and apple
+
+    # Grab reference to apple and knife
+    apple = og.sim.scene.object_registry("name", "apple")
+    knife = og.sim.scene.object_registry("name", "knife")
+    place_obj_on_floor_plane(apple)
+    place_obj_on_floor_plane(knife)
+
+
+    # Let apple settle
+    for _ in range(50):
+        og.sim.step()
+
+    # knife.keep_still()
+    knife.set_position_orientation(
+        position=apple.get_position() + np.array([-0.15, 0.0, 0.2]),
+        orientation=T.euler2quat([-np.pi / 4, 0, 0]),
+    )
+
+    # Step simulation for a bit so that apple is sliced
+    for i in range(10000):
+        og.sim.step()
+
+    # Check if apple is sliced
+    is_empty = len(og.sim.scene.object_registry("category", "half_apple", default_val=[])) == 0 
+    assert not is_empty
+
+    # Remove objects
+    og.sim.remove_object(apple)
+    og.sim.remove_object(knife)
+
+@og_test
+def test_slicer_volume_blunt_end():
+    """
+    Demo of slicing an apple into two apple slices
+    """
+
+    # Create the scene config to load -- empty scene with knife and apple
+
+    # Grab reference to apple and knife
+    apple = og.sim.scene.object_registry("name", "apple")
+    knife = og.sim.scene.object_registry("name", "knife")
+    place_obj_on_floor_plane(apple)
+    place_obj_on_floor_plane(knife)
+
+    # Let apple settle
+    for _ in range(50):
+        og.sim.step()
+
+    knife.keep_still()
+    knife.set_position_orientation(
+        position=apple.get_position() + np.array([-0.15, 0.0, 0.15]),
+        orientation=T.euler2quat([np.pi / 2, 0, 0]),
+    )
+
+    # Step simulation for a bit so that apple is sliced
+    for i in range(100):
+        og.sim.step()
+
+
+    # Check if apple is not sliced
+    is_sliced = og.sim.scene.object_registry("name", "apple") is None
+    assert not is_sliced
+
+    # Remove objects
+    og.sim.remove_object(apple)
+    og.sim.remove_object(knife)
+
+if __name__ == "__main__":
+    test_slicer_volume_sharp_end()
\ No newline at end of file