Address PR feedback: simplify tests based on reviewer suggestions

Co-authored-by: IvanYashchuk <[email protected]>

Author: Copilot

tests/cpp/test_alias.cpp

@@ -1670,9 +1670,12 @@ TEST_F(AliasTest, BroadcastInDimNoRedundantSet) {
   fusion->addInput(in);
 
   // Call broadcast with all dims marked as non-broadcast
-  // This should not introduce a Set operation
+  // This should not introduce a Set operation and return the input directly
   std::vector<bool> is_broadcast_dim = {false, false};
-  TensorView* out = broadcast(in, is_broadcast_dim);
+  TensorView* maybe_bcast = broadcast(in, is_broadcast_dim);
+  
+  // Add an operation to ensure we have something to test
+  TensorView* out = abs(maybe_bcast);
 
   fusion->addOutput(out);
 

tests/python/test_python_frontend.py

@@ -707,20 +707,9 @@ def fusion_with_expand(fd: FusionDefinition):
         self.assertEqual(eager_out, nvf_out_exp[0])
 
         # Check that the broadcast_in_dim fusion doesn't have a redundant Set operation
-        # by comparing the IR string representations
-        bid_str = str(fd_bid)
-        exp_str = str(fd_exp)
-        
-        # Count tensor cast operations (not scalar casts)
-        bid_tensor_casts = bid_str.count("fd.ops.cast(t")
-        exp_tensor_casts = exp_str.count("fd.ops.cast(t")
-        
-        # They should have the same number of tensor casts
-        self.assertEqual(
-            bid_tensor_casts,
-            exp_tensor_casts,
-            f"broadcast_in_dim has {bid_tensor_casts} tensor casts but expand has {exp_tensor_casts}"
-        )
+        # by comparing the IR string representations - they should be identical since
+        # broadcast is a no-op in this case
+        self.assertEqual(str(fd_bid), str(fd_exp))
 
     # Testing a scenario where the broadcast is necessary to realize the output
     def test_tensor_shape_with_output_bcast(self):