def test_fixed_training_size(self):
data = Table("iris")
train, test = sample(data, 100)
self.assertEqual(len(train), 100)
self.assertEqual(len(train) + len(test), len(data))
train, test = sample(data, 0.1)
self.assertEqual(len(train), 15)
self.assertEqual(len(train) + len(test), len(data))
train, test = sample(data, 0.1, stratified=True)
self.assertEqual(len(train), 15)
self.assertEqual(len(train) + len(test), len(data))
train, test = sample(data, 0.2, replace=True)
self.assertEqual(len(train), 30)
train, test = sample(data, 0.9, replace=True)
self.assertEqual(len(train), 135)
self.assertGreater(len(train) + len(test), len(data))
def test_fixed_training_size(self):
data = Table("iris")
train, test = sample(data, 100)
self.assertEqual(len(train), 100)
self.assertEqual(len(train) + len(test), len(data))
train, test = sample(data, 0.1)
self.assertEqual(len(train), 15)
self.assertEqual(len(train) + len(test), len(data))
train, test = sample(data, 0.1, stratified=True)
self.assertEqual(len(train), 15)
self.assertEqual(len(train) + len(test), len(data))
train, test = sample(data, 0.2, replace=True)
self.assertEqual(len(train), 30)
train, test = sample(data, 0.9, replace=True)
self.assertEqual(len(train), 135)
self.assertGreater(len(train) + len(test), len(data))