Commits

Andrej T. committed 7bb2fe9

Updated test-imputation.py unit test.

Comments (0)

Files changed (1)

Orange/testing/unit/tests/test_imputation.py

 	@test_on_data
 	def test_minimal(self, dataset):
 		imputer = MinimalConstructor(dataset)
-		print "Min", imputer(dataset[1])[2] == 1 # Make this an AssertTrue
+		self.assertEquals(imputer(dataset[1])[2], 1)
 
 	@test_on_data
 	def test_maximal(self, dataset):
 		imputer = MaximalConstructor(dataset)
-		print "Max", imputer(dataset[1])[2] == 3 # Make this an AssertTrue
+		self.assertEquals(imputer(dataset[1])[2], 3)
 
 	@test_on_data
 	def test_average(self, dataset):
 		imputer = AverageConstructor(dataset)
-		print "Ave", imputer(dataset[1])[2] == 2 # Make this an AssertTrue
+		self.assertEquals(imputer(dataset[1])[2], 2)
+		
 if __name__ == "__main__":
 	unittest.main()