summaryrefslogtreecommitdiff
path: root/ext/-test-/method/arity.c
blob: 239b9f7f10135d9e4cd318b55f6673fdd4c77d9f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include "ruby.h"

static VALUE
obj_method_arity(VALUE self, VALUE obj, VALUE mid)
{
    int arity = rb_obj_method_arity(obj, rb_check_id(&mid));
    return INT2FIX(arity);
}

static VALUE
mod_method_arity(VALUE self, VALUE mod, VALUE mid)
{
    int arity = rb_mod_method_arity(mod, rb_check_id(&mid));
    return INT2FIX(arity);
}

void
Init_arity(VALUE mod)
{
    rb_define_module_function(mod, "obj_method_arity", obj_method_arity, 2);
    rb_define_module_function(mod, "mod_method_arity", mod_method_arity, 2);
}