More test cases for more types of mehtods

deephaven · Mar 10, 2023 · a35bffc · a35bffc
1 parent 8836912
commit a35bffc
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/py/server/tests/test_table.py b/py/server/tests/test_table.py
@@ -918,7 +918,15 @@ class Foo:
  def __call__(self):
  ...
 
- def do_something(self, p=None):
+ def do_something_instance(self, p=None):
+ return p if p else 1
+
+ @classmethod
+ def do_something_cls(cls, p=None):
+ return p if p else 1
+
+ @staticmethod
+ def do_something_static(p=None):
  return p if p else 1
 
  def do_something(p=None):
@@ -931,7 +939,13 @@ def do_something(p=None):
  self.assertTrue(rt.columns[0].data_type == dtypes.int32)
 
  foo = Foo()
- rt = empty_table(1).update("Col = (int)foo.do_something()")
+ rt = empty_table(1).update("Col = (int)foo.do_something_instance()")
+ self.assertTrue(rt.columns[0].data_type == dtypes.int32)
+
+ rt = empty_table(1).update("Col = (int)Foo.do_something_cls()")
+ self.assertTrue(rt.columns[0].data_type == dtypes.int32)
+
+ rt = empty_table(1).update("Col = (int)foo.do_something_static()")
  self.assertTrue(rt.columns[0].data_type == dtypes.int32)
 
  rt = empty_table(1).update("Col = (int)do_something((byte)Foo.ATTR)")