def test_four(self):
labels, truth = create_sim_dataset(1000, 6, 3, 0.8, 0.8, True, False,
False)
iwmv_obj = IWMV()
result = iwmv_obj.fit(labels)
accuracy = result['accuracies']
self.assertTrue(np.all(np.abs(accuracy - truth) < 0.075))
def test_return_fields(self):
iwmv_obj = IWMV()
result = iwmv_obj.fit(MED_MATRIX)
self.assertEqual(set(result.keys()), FIT_RETURN_SET)