diff options
Diffstat (limited to 'spec/ruby/optional/capi/ext/enumerator_spec.c')
| -rw-r--r-- | spec/ruby/optional/capi/ext/enumerator_spec.c | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/spec/ruby/optional/capi/ext/enumerator_spec.c b/spec/ruby/optional/capi/ext/enumerator_spec.c new file mode 100644 index 0000000000..917621c003 --- /dev/null +++ b/spec/ruby/optional/capi/ext/enumerator_spec.c @@ -0,0 +1,32 @@ +#include "ruby.h" +#include "rubyspec.h" + +#ifdef __cplusplus +extern "C" { +#endif + +VALUE enumerator_spec_rb_enumeratorize(int argc, VALUE *argv, VALUE self) { + VALUE obj, meth, args; + rb_scan_args(argc, argv, "2*", &obj, &meth, &args); + return rb_enumeratorize(obj, meth, (int)RARRAY_LEN(args), RARRAY_PTR(args)); +} + +VALUE enumerator_spec_size_fn(VALUE obj, VALUE args, VALUE anEnum) { + return INT2NUM(7); +} + +VALUE enumerator_spec_rb_enumeratorize_with_size(int argc, VALUE *argv, VALUE self) { + VALUE obj, meth, args; + rb_scan_args(argc, argv, "2*", &obj, &meth, &args); + return rb_enumeratorize_with_size(obj, meth, (int)RARRAY_LEN(args), RARRAY_PTR(args), enumerator_spec_size_fn); +} + +void Init_enumerator_spec(void) { + VALUE cls = rb_define_class("CApiEnumeratorSpecs", rb_cObject); + rb_define_method(cls, "rb_enumeratorize", enumerator_spec_rb_enumeratorize, -1); + rb_define_method(cls, "rb_enumeratorize_with_size", enumerator_spec_rb_enumeratorize_with_size, -1); +} + +#ifdef __cplusplus +} +#endif |
