Add test case for in method

Author: orangeH25

test/test_utils.py

@@ -144,6 +144,10 @@ def _test_default_impls_helper(self, lp_tensor, lp_tensor_for_copy):
         self.assertTrue(torch.equal(lp_tensor.qdata, reconstructed.qdata))
         self.assertEqual(lp_tensor.attr, reconstructed.attr)
 
+        # test _get_to_kwargs
+        _ = lp_tensor._get_to_kwargs(torch.strided, device="cuda")
+        _ = lp_tensor._get_to_kwargs(layout=torch.strided, device="cuda")
+
         # `to` / `_to_copy`
         original_device = lp_tensor.device
         lp_tensor = lp_tensor.to("cuda")
@@ -340,9 +344,6 @@ def __init__(
         )
         self._test_default_impls_helper(lp_tensor, lp_tensor_for_copy)
 
-    def test__get_to_kwargs_with_layout(self):
-        MyClass = TorchAOBaseTensor()
-        MyClass._get_to_kwargs(torch.strided, device="cuda")
 
 if __name__ == "__main__":
     unittest.main()