11const ArraySet = require ( './array-set' ) ;
22const HashSet = require ( './hash-set' ) ;
3- const HashMapSet = require ( './hash-set-1' ) ;
3+ const MapSet = require ( './map-set' ) ;
4+ const TreeSet = require ( './tree-set' ) ;
45
5- [ HashSet , HashMapSet , ArraySet ] . forEach ( ( MySet ) => {
6+ // const setImplementations = [HashSet, MapSet, ArraySet, TreeSet, Set];
7+ const setImplementations = [ Set , HashSet , TreeSet ] ;
8+
9+ setImplementations . forEach ( ( MySet ) => {
610 describe ( `Set (${ MySet . name } )` , ( ) => {
711 let set ;
812
@@ -43,7 +47,7 @@ const HashMapSet = require('./hash-set-1');
4347 set . add ( 1 ) ;
4448 set . add ( 2 ) ;
4549 set . add ( 3 ) ;
46- expect ( set . entries ( ) ) . toEqual ( [ 1 , 2 , 3 ] ) ;
50+ expect ( [ ... set . entries ( ) ] ) . toEqual ( [ [ 1 , 1 ] , [ 2 , 2 ] , [ 3 , 3 ] ] ) ;
4751 } ) ;
4852
4953 it ( 'should return entries wihout holes' , ( ) => {
@@ -53,17 +57,17 @@ const HashMapSet = require('./hash-set-1');
5357 set . add ( 3 ) ;
5458
5559 expect ( set . delete ( 2 ) ) . toBe ( true ) ;
56- expect ( set . entries ( ) ) . toEqual ( [ 0 , 1 , 3 ] ) ;
60+ expect ( Array . from ( set . entries ( ) ) ) . toEqual ( [ [ 0 , 0 ] , [ 1 , 1 ] , [ 3 , 3 ] ] ) ;
5761 expect ( set . delete ( 0 ) ) . toBe ( true ) ;
5862
59- expect ( set . entries ( ) ) . toEqual ( [ 1 , 3 ] ) ;
63+ expect ( Array . from ( set ) ) . toEqual ( [ 1 , 3 ] ) ;
6064 expect ( set . size ) . toBe ( 2 ) ;
6165 } ) ;
6266
6367 it ( 'should initialize with data provided' , ( ) => {
6468 set = new MySet ( [ 1 , 2 , 3 , 1 ] ) ;
6569 expect ( set . size ) . toBe ( 3 ) ;
66- expect ( set . entries ( ) ) . toEqual ( [ 1 , 2 , 3 ] ) ;
70+ expect ( Array . from ( set . keys ( ) ) ) . toEqual ( [ 1 , 2 , 3 ] ) ;
6771 } ) ;
6872
6973 it ( 'should return an iterable' , ( ) => {
0 commit comments