def test_store__fetch__remove(self):
d = Dataset().data
self.db.store(d)
self.assertEqual(d, self.db.fetch(d.id, lazy=False))
# self.assertEqual(d, self.db.fetch(d.id, lazy=True))
self.assertEqual(d, self.db.fetch(d))
self.db.removedata(d)
self.assertEqual(None, self.db.fetch(d))
def test__rus_(self):
augmented_iris = Root >> (
Dataset() * ROS_(strategy={
"virginica": 100,
"versicolor": 50,
"setosa": 50
}))
self.assertEqual(150, len((augmented_iris >> RUS_).X))
def test__rus(self):
augmented_iris = Root >> (
Dataset() * ROS_(strategy={
"virginica": 100,
"versicolor": 50,
"setosa": 50
}))
self.assertEqual(200, len(
(augmented_iris >> AutoIns * RUS).X)) # TODO: inhibit UserWarning
self.assertEqual(150, len((augmented_iris >> AutoIns * RUS).inner.X))
def setUp(self):
warnings.simplefilter('ignore',
(DeprecationWarning, UserWarning, ImportWarning))
self.db = Tatu("sqlite://:memory:", threaded=False)
d = Dataset().data
d["X"] = [[1, 2, 3]]
d["y"] = [1, 2, 3]
self.d = d
self.db.store(d)
self.letstep = Let(field="X",
value=[[1, 2,
3]]) # * Let(field="y", value=[1, 2, 3])
def test__ros(self):
iris = Root >> (Dataset() * AutoIns)
truncated_iris = iris >> Slice(last=119)
rebalanced_iris = truncated_iris >> ROS
self.assertEqual(
50,
DataFrame(rebalanced_iris.inner.y).value_counts()["virginica"])
larger_iris = iris >> ROS(strategy={
"virginica": 100,
"versicolor": 50,
"setosa": 50
})
self.assertEqual(200, len(larger_iris.inner.X))
def test__process_(self):
truncated_iris = Root >> (Dataset() * Slice(code=":120"))
self.assertEqual(
20,
DataFrame(truncated_iris.y).value_counts()["virginica"])
def test__process_(self):
d = Dataset().data
self.assertEqual("19b2d27779bc2d2444c11f5cc24c98ee", mathash("X", d.X))
self.assertEqual("8baa54c6c205d73f99bc1215b7d46c9c", mathash("y", d.y))
def test__rus_(self):
samplediris = Root >> (Dataset() * Sample_(n=120))
self.assertEqual(120, len(samplediris.X))
def test__rus(self):
samplediris = Root >> (Dataset() * Sample_(n=140))
self.assertEqual(140, len(
(samplediris >> AutoIns * Sample).X)) # TODO: inhibit UserWarning
self.assertEqual(100, len((samplediris >> AutoIns * Sample).inner.X))