import unittest

from src.ex1_triangles import countTriangles


class MyTestCase(unittest.TestCase):
    def test_simple1(self):
        self.assertEqual(
            countTriangles([(0, 1), (1, 2), (0, 2)]),
            1
        )

    def test_simple2(self):
        self.assertEqual(
            countTriangles([(2, 1), (1, 2), (2, 3)]),
            0
        )


if __name__ == '__main__':
    unittest.main()