def test_validate_size(): train_size, val_size, test_size = RatioSplit.validate_size(0.1, 0.2, 10) assert 7 == train_size assert 1 == val_size assert 2 == test_size train_size, val_size, test_size = RatioSplit.validate_size(None, 0.5, 10) assert 5 == train_size assert 0 == val_size assert 5 == test_size train_size, val_size, test_size = RatioSplit.validate_size(None, None, 10) assert 10 == train_size assert 0 == val_size assert 0 == test_size train_size, val_size, test_size = RatioSplit.validate_size(2, 2, 10) assert 6 == train_size assert 2 == val_size assert 2 == test_size try: RatioSplit.validate_size(-1, 0.2, 10) except ValueError: assert True try: RatioSplit.validate_size(1, -0.2, 10) except ValueError: assert True try: RatioSplit.validate_size(11, 0.2, 10) except ValueError: assert True try: RatioSplit.validate_size(0, 11, 10) except ValueError: assert True try: RatioSplit.validate_size(3, 8, 10) except ValueError: assert True
def test_validate_size(self): train_size, val_size, test_size = RatioSplit.validate_size(0.1, 0.2, 10) self.assertEqual(train_size, 7) self.assertEqual(val_size, 1) self.assertEqual(test_size, 2) train_size, val_size, test_size = RatioSplit.validate_size(None, 0.5, 10) self.assertEqual(train_size, 5) self.assertEqual(val_size, 0) self.assertEqual(test_size, 5) train_size, val_size, test_size = RatioSplit.validate_size(None, None, 10) self.assertEqual(train_size, 10) self.assertEqual(val_size, 0) self.assertEqual(test_size, 0) train_size, val_size, test_size = RatioSplit.validate_size(2, 2, 10) self.assertEqual(train_size, 6) self.assertEqual(val_size, 2) self.assertEqual(test_size, 2) try: RatioSplit.validate_size(-1, 0.2, 10) except ValueError: assert True try: RatioSplit.validate_size(1, -0.2, 10) except ValueError: assert True try: RatioSplit.validate_size(11, 0.2, 10) except ValueError: assert True try: RatioSplit.validate_size(0, 11, 10) except ValueError: assert True try: RatioSplit.validate_size(3, 8, 10) except ValueError: assert True