Merge pull request #497 from cptobvious/string-ord

added ord method to string object
topazproject · Mar 10, 2013 · c5dbd88 · c5dbd88
2 parents fba2389 + 7cdaefd
commit c5dbd88
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/tests/objects/test_stringobject.py b/tests/objects/test_stringobject.py
@@ -36,6 +36,14 @@ def test_to_str(self, space):
         w_res = space.execute('return "ABC".to_str')
         assert space.str_w(w_res) == "ABC"
 
+    def test_ord(self, space):
+        w_res = space.execute('return "a".ord')
+        assert space.int_w(w_res) == 97
+        w_res = space.execute('return "asd".ord')
+        assert space.int_w(w_res) == 97
+        with self.raises(space, "ArgumentError", "empty string"):
+            space.execute("''.ord")
+
     def test_length(self, space):
         w_res = space.execute("return 'ABC'.length")
         assert space.int_w(w_res) == 3

diff --git a/topaz/objects/stringobject.py b/topaz/objects/stringobject.py
@@ -414,6 +414,13 @@ def method_initialize_copy(self, space, w_other):
     def method_to_s(self, space):
         return self
 
+    @classdef.method("ord")
+    def method_ord(self, space):
+        string = space.str_w(self)
+        if len(string) == 0:
+            raise space.error(space.w_ArgumentError, "empty string")
+        return space.newint(ord(string[0]))
+
     @classdef.method("inspect")
     def method_inspect(self, space):
         return space.newstr_fromstr('"%s"' % self.str_w(space))