1#include "ruby.h"
2
3static VALUE
4obj_method_arity(VALUE self, VALUE obj, VALUE mid)
5{
6    int arity = rb_obj_method_arity(obj, rb_check_id(&mid));
7    return INT2FIX(arity);
8}
9
10static VALUE
11mod_method_arity(VALUE self, VALUE mod, VALUE mid)
12{
13    int arity = rb_mod_method_arity(mod, rb_check_id(&mid));
14    return INT2FIX(arity);
15}
16
17void
18Init_arity(VALUE mod)
19{
20    rb_define_module_function(mod, "obj_method_arity", obj_method_arity, 2);
21    rb_define_module_function(mod, "mod_method_arity", mod_method_arity, 2);
22}
23