def test_sexmachine_sgd_load(self):
s = DameSexmachine()
m = s.sgd_load()
predicted = m.predict([[
0, 0, 1, 0, 21, 0, 0, 0, 0, 34, 2, 0, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0,
0, 0, 0, 2, 0, 0, 0, 34, 1, 0, 1
]])
n = np.array([1])
self.assertEqual(n, predicted)
def test_sexmachine_sgd_load_method_returns_correct_result(self):
s = DameSexmachine()
m = s.sgd_load()
predicted = m.predict([[ 0, 0, 1, 0, 21, 0, 0, 0, 0, 34, 2, 0, 0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 2, 0, 0, 0, 34, 1, 0, 1, 0]])
n = np.array([2])
self.assertEqual(n, predicted)
