Add tests for geometry/entity

harryzcy · harryzcy · commit 0da76c7fd674 · 2020-10-15T02:06:27.000+08:00
diff --git a/sympy/geometry/tests/test_entity.py b/sympy/geometry/tests/test_entity.py
@@ -1,11 +1,31 @@
 from sympy import Symbol, Rational
 from sympy.geometry import Circle, Ellipse, Line, Point, Polygon, Ray, RegularPolygon, Segment, Triangle
-from sympy.geometry.entity import scale
+from sympy.geometry.entity import scale, GeometryEntity
 from sympy.testing.pytest import raises
 
 from random import random
 
 
+def test_entity():
+    x = Symbol('x', real=True)
+    y = Symbol('y', real=True)
+
+    assert GeometryEntity(x, y) in GeometryEntity(x, y)
+    raises(NotImplementedError, lambda: Point(0, 0) in GeometryEntity(x, y))
+
+    assert GeometryEntity(x, y) == GeometryEntity(x, y)
+    assert GeometryEntity(x, y).equals(GeometryEntity(x, y))
+
+    c = Circle((0, 0), 5)
+    assert GeometryEntity.encloses(c, Point(0, 0))
+    assert GeometryEntity.encloses(c, Segment((0, 0), (1, 1)))
+    assert GeometryEntity.encloses(c, Line((0, 0), (1, 1))) is False
+    assert GeometryEntity.encloses(c, Circle((0, 0), 4))
+    assert GeometryEntity.encloses(c, Polygon(Point(0, 0), Point(1, 0), Point(0, 1)))
+    assert GeometryEntity.encloses(c, RegularPolygon(Point(8, 8), 1, 3)) is False
+
+
+
 def test_subs():
     x = Symbol('x', real=True)
     y = Symbol('y', real=True)
