Implemented __len__ for sets, frozensets and dicts

kunparekh18 · May 25, 2017 · 9b9b495 · 9b9b495
1 parent eeffcf1
commit 9b9b495
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 4 deletions.
diff --git a/batavia/types/Dict.js b/batavia/types/Dict.js
@@ -106,6 +106,10 @@ Dict.prototype.toString = function() {
  * Type conversions
  **************************************************/
 
+Dict.prototype.__len__ = function() {
+ return this.size
+}
+
 Dict.prototype.__bool__ = function() {
  return this.size > 0
 }

diff --git a/batavia/types/FrozenSet.js b/batavia/types/FrozenSet.js
@@ -34,6 +34,10 @@ FrozenSet.prototype.toString = function() {
  * Type conversions
  **************************************************/
 
+FrozenSet.prototype.__len__ = function() {
+ return this.data.size
+}
+
 FrozenSet.prototype.__bool__ = function() {
  return this.data.__bool__()
 }

diff --git a/batavia/types/Set.js b/batavia/types/Set.js
@@ -34,6 +34,10 @@ Set.prototype.toString = function() {
  * Type conversions
  **************************************************/
 
+Set.prototype.__len__ = function() {
+ return this.data.size
+}
+
 Set.prototype.__bool__ = function() {
  return this.data.__bool__()
 }

diff --git a/tests/builtins/test_len.py b/tests/builtins/test_len.py
@@ -18,7 +18,4 @@ class BuiltinLenFunctionTests(BuiltinFunctionTestCase, TranspileTestCase):
 
  not_implemented = [
  'test_bytearray',
- 'test_dict',
- 'test_frozenset',
- 'test_set',
  ]
diff --git a/tests/structures/test_set_comprehension.py b/tests/structures/test_set_comprehension.py
@@ -4,7 +4,6 @@
 
 
 class SetComprehensionTests(TranspileTestCase):
- @unittest.expectedFailure
  def test_syntax(self):
  self.assertCodeExecution("""
  x = [1, 2, 3, 4, 5]