@@ -78,7 +78,7 @@ def test_drop_remainder_true(self):
7878 drop_remainder = True ,
7979 shuffle = False ,
8080 )
81- self .assertEquals (data_generator .batches_per_epoch , 3 )
81+ self .assertEqual (data_generator .batches_per_epoch , 3 )
8282 for i in range (3 ):
8383 data_generator .get_batch ()
8484 _ , y_batch = data_generator .get_batch ()
@@ -92,11 +92,11 @@ def test_nothing_to_drop(self):
9292 drop_remainder = True ,
9393 shuffle = False ,
9494 )
95- self .assertEquals (data_generator .batches_per_epoch , 10 )
95+ self .assertEqual (data_generator .batches_per_epoch , 10 )
9696 for i in range (9 ):
9797 data_generator .get_batch ()
9898 _ , y_batch = data_generator .get_batch ()
99- self .assertEquals (len (y_batch ), 10 )
99+ self .assertEqual (len (y_batch ), 10 )
100100
101101 def test_drop_remainder_false (self ):
102102 data_generator = NumpyDataGenerator (
@@ -106,11 +106,11 @@ def test_drop_remainder_false(self):
106106 drop_remainder = False ,
107107 shuffle = False ,
108108 )
109- self .assertEquals (data_generator .batches_per_epoch , 4 )
109+ self .assertEqual (data_generator .batches_per_epoch , 4 )
110110 for i in range (3 ):
111111 data_generator .get_batch ()
112112 x_batch , _ = data_generator .get_batch ()
113- self .assertEquals (len (x_batch ), self .m % self .batch_size )
113+ self .assertEqual (len (x_batch ), self .m % self .batch_size )
114114
115115 def test_shuffle (self ):
116116 """
@@ -143,7 +143,7 @@ def test_single_epoch_generator(self):
143143 drop_remainder = True ,
144144 shuffle = False ,
145145 )
146- self .assertEquals (data_generator .batches_per_epoch , 3 )
146+ self .assertEqual (data_generator .batches_per_epoch , 3 )
147147 for i in range (3 ):
148148 data_generator .get_batch ()
149149 _ , y_batch = data_generator .get_batch ()
0 commit comments