Add pytorch tests

Author: ElleNajt

tests/babel-formatting.org

@@ -486,3 +486,44 @@ print(x)
   │   │   [-0.6745,  0.0472, -0.8954]])
   :end:
 
+
+  
+  #+begin_src python
+import torch
+
+# Test tensor
+x = torch.randn(3, 3)
+print("Tensor:", x)
+# Test dict
+d = {"name": "Alice", "age": 30, "city": "New York", "hobbies": ["reading", "coding"]}
+print("Dict:", d)
+
+# Test list
+l = [1, 2, 3, [4, 5, 6], {"nested": "dict"}]
+print("List:", l)
+
+# Test set
+s = {1, 2, 3, 4, 5}
+print("Set:", s)
+import os
+os.path
+print(100)
+  #+end_src
+
+  #+RESULTS:
+  :results:
+  Tensor:
+  tensor([[-1.3200,  1.4083,  1.1319],
+  │   │   [-0.6430, -1.6252, -1.3978],
+  │   │   [ 1.0347,  1.5210,  1.5890]])
+  Dict:
+  {'name': 'Alice', 'age': 30, 'city': 'New York', 'hobbies': ['reading', 'coding']}
+  List:
+  [1, 2, 3, [4, 5, 6], {'nested': 'dict'}]
+  Set:
+  {1, 2, 3, 4, 5}
+  100
+  :end:
+
+
+