class ModelTests(unittest.TestCase):
def setUp(self):
self.model = LinearMRF(3, 2)
def test_unary_feature_shape(self):
img = np.array([1, 0, 0, 1, 0, 1])
result = self.model.get_unary_features(img)
self.assertEqual(result.shape, (6, 2))
def test_pairwise_feature_shape(self):
result = self.model.get_pairwise_features()
self.assertEqual(result.shape, (7, 4))
class ModelTests(unittest.TestCase):
def setUp(self):
self.model = LinearMRF(3, 2)
def test_unary_feature_shape(self):
img = np.array([1, 0, 0, 1, 0, 1])
result = self.model.get_unary_features(img)
self.assertEqual(result.shape, (6, 2))
def test_unary_feature_data(self):
img = np.array([1, 0, 0, 1, 0, 1])
result = self.model.get_unary_features(img)
expected = np.array([[0, 1], [1, 0], [1, 0], [0, 1], [1, 0], [0, 1]])
np.testing.assert_array_equal(result, expected)
def test_pairwise_feature_shape(self):
result = self.model.get_pairwise_features()
self.assertEqual(result.shape, (7, 4))
def test_pairwise_feature_data(self):
result = self.model.get_pairwise_features()
np.testing.assert_array_equal(result, [[1, 0, 0, 1]] * 7)