.. -*- coding: utf-8 -*-
.. _coercion_and_categories:
=================================================
How to implement new algebraic structures in Sage
=================================================
--------------------------------------
Sage's category and coercion framework
--------------------------------------
.. MODULEAUTHOR::
Simon King,
Friedrich\--Schiller\--Universität Jena,
© 2011/2013
.. toctree::
:maxdepth: 2
.. linkall
The aim of this tutorial is to explain how one can benefit from Sage's
category framework and coercion model when implementing new algebraic
structures. It is based on a worksheet created in 2011.
We illustrate the concepts of Sage's category framework and coercion model by
means of a detailed example, namely a toy implementation of fraction fields.
The code is developed step by step, so that the reader can focus on one detail
in each part of this tutorial. The complete code can be found in the appendix.
Outline
=======
- Use existing base classes
For using Sage's coercion system, it is essential to work with sub\--classes
of :class:`sage.structure.parent.Parent` or
:class:`sage.structure.element.Element`, respectively. They provide default
implementations of many "magical" double-underscore Python methods, which
must not be overridden. Instead, the actual implementation should be in
*single underscore* methods, such as ``_add_`` or ``_mul_``.
- Turn your parent structure into an object of a category
Declare the category during initialisation\---Your parent structure will
inherit further useful methods and consistency tests.
- Provide your parent structure with an element class
Assign to it an attribute called ``Element``\---The elements will inherit
further useful methods from the category. In addition, some basic
conversions will immediately work.
- Implement further conversions
Never override a parent's ``__call__`` method! Provide the method
``_element_constructor_`` instead.
- Declare coercions
If a conversion happens to be a morphism, you may consider to turn it into a
coercion. It will then *implicitly* be used in arithmetic operations.
- Advanced coercion: Define construction functors for your parent structure
Sage will automatically create new parents for you when needed, by the
so\--called :func:`sage.categories.pushout.pushout` construction.
- Run the automatic test suites
Each method should be documented and provide a doc test (we are not giving
examples here). In addition, any method defined for the objects or elements
of a category should be supported by a test method, that is executed when
running the test suite.
Base classes
============
In Sage, a "Parent" is an object of a category and contains elements. Parents
should inherit from :class:`sage.structure.parent.Parent` and their elements
from :class:`sage.structure.element.Element`.
Sage provides appropriate sub\--classes of
:class:`~sage.structure.parent.Parent` and
:class:`~sage.structure.element.Element` for a variety of more concrete
algebraic structures, such as groups, rings, or fields, and of their
elements. But some old stuff in Sage doesn't use it. **Volunteers for
refactoring are welcome!**
The parent
----------
Since we wish to implement a special kind of fields, namely fraction fields,
it makes sense to build on top of the base class
:class:`sage.rings.ring.Field` provided by Sage. ::
sage: from sage.rings.ring import Field
This base class provides a lot more methods than a general parent::
sage: [p for p in dir(Field) if p not in dir(Parent)]
['__div__', '__fraction_field', '__ideal_monoid', '__iter__',
'__pow__', '__rdiv__', '__rpow__', '__rxor__', '__xor__',
'_an_element', '_an_element_c', '_an_element_impl', '_coerce_',
'_coerce_c', '_coerce_impl', '_coerce_self', '_coerce_try',
'_default_category', '_gens', '_gens_dict',
'_has_coerce_map_from', '_ideal_class_', '_latex_names', '_list',
'_one_element', '_pseudo_fraction_field',
'_random_nonzero_element', '_richcmp', '_unit_ideal',
'_zero_element', '_zero_ideal', 'algebraic_closure',
'base_extend', 'cardinality', 'class_group', 'coerce_map_from_c',
'coerce_map_from_impl', 'content', 'divides', 'extension',
'fraction_field', 'frobenius_endomorphism', 'gcd', 'gen', 'gens',
'get_action_c', 'get_action_impl', 'has_coerce_map_from_c',
'has_coerce_map_from_impl', 'ideal', 'ideal_monoid',
'integral_closure', 'is_commutative', 'is_field', 'is_finite',
'is_integral_domain', 'is_integrally_closed', 'is_noetherian',
'is_prime_field', 'is_ring', 'is_subring',
'krull_dimension', 'list', 'ngens', 'one', 'one_element',
'order', 'prime_subfield', 'principal_ideal', 'quo', 'quotient',
'quotient_ring', 'random_element', 'unit_ideal', 'zero',
'zero_element', 'zero_ideal', 'zeta', 'zeta_order']
The following is a very basic implementation of fraction fields, that needs to
be complemented later.
::
sage: from sage.structure.unique_representation import UniqueRepresentation
sage: class MyFrac(UniqueRepresentation, Field):
....: def __init__(self, base):
....: if base not in IntegralDomains():
....: raise ValueError, "%s is no integral domain"%base
....: Field.__init__(self, base)
....: def _repr_(self):
....: return "NewFrac(%s)"%repr(self.base())
....: def base_ring(self):
....: return self.base().base_ring()
....: def characteristic(self):
....: return self.base().characteristic()
.. end ouf output
This basic implementation is formed by the following steps:
- Any ring in Sage has a **base** and a **base ring**. The "usual" fraction
field of a ring `R` has the base `R` and the base ring ``R.base_ring()``::
sage: Frac(QQ['x']).base(), Frac(QQ['x']).base_ring()
(Univariate Polynomial Ring in x over Rational Field, Rational Field)
Declaring the base is easy: We just pass it as an argument to the field
constructor.
::
sage: Field(ZZ['x']).base()
Univariate Polynomial Ring in x over Integer Ring
.. end of output
We are implementing a seperate method returning the base ring.
- Python uses double\--underscore methods for arithemetic methods and string
representations. Sage's base classes often have a default implementation,
and it is requested to **implement SINGLE underscore methods _repr_, and
similarly _add_, _mul_ etc.**
- You are encouraged to **make your parent "unique"**. That's to say, parents
should only evaluate equal if they are identical. Sage provides frameworks
to create unique parents. We use here the most easy one: Inheriting from the
class :class:`sage.structure.unique_representation.UniqueRepresentation` is
enough. Making parents unique can be quite important for an efficient
implementation, because the repeated creation of "the same" parent would
take a lot of time.
- Fraction fields are only defined for integral domains. Hence, we raise an
error if the given ring does not belong to the category of integral
domains. This is our first use case of categories.
- Last, we add a method that returns the characteristic of the field. We don't
go into details, but some automated tests that we study below implicitly
rely on this method.
We see that our basic implementation correctly refuses a ring that is not an
integral domain::
sage: MyFrac(ZZ['x'])
NewFrac(Univariate Polynomial Ring in x over Integer Ring)
sage: MyFrac(Integers(15))
Traceback (most recent call last):
...
ValueError: Ring of integers modulo 15 is no integral domain
.. NOTE::
Inheritance from :class:`~sage.structure.unique_representation.UniqueRepresentation`
automatically provides our class with pickling, preserving the unique
parent condition. If we had defined the class in some external module or
in an interactive session, pickling would work immediately.
However, for making the following example work in Sage's doctesting
framework, we need to assign our class as an attribute of the ``__main__``
module, so that the class can be looked up during unpickling.
::
sage: import __main__
sage: __main__.MyFrac = MyFrac
sage: loads(dumps(MyFrac(ZZ))) is MyFrac(ZZ)
True
.. NOTE::
In the following sections, we will successively add or change details of
``MyFrac``. Rather than giving a full class definition in each step, we
define new versions of ``MyFrac`` by inheriting from the previously
defined version of ``MyFrac``. We believe this will help the reader to
focus on the single detail that is relevant in each section.
The complete code can be found in the appendix.
The elements
------------
We use the base class :class:`sage.structure.element.FieldElement`. Note that
in the creation of field elements it is not tested that the given parent is a
field::
sage: from sage.structure.element import FieldElement
sage: FieldElement(ZZ)
Generic element of a structure
Our toy implementation of fraction field elements is based on the following
considerations:
- A fraction field element is defined by numerator and denominator, which both
need to be elements of the base. There should be methods returning numerator
resp. denominator.
- The denominator must not be zero, and (provided that the base is an ordered
ring) we can make it non-negative, without loss of generality. By default,
the denominator is one.
- The string representation is returned by the single\--underscore method
``_repr_``. In order to make our fraction field elements distinguishable
from those already present in Sage, we use a different string representation.
- Arithmetic is implemented in single\--underscore method ``_add_``, ``_mul_``,
etc. **We do not override the default double underscore __add__, __mul__**,
since otherwise, we could not use Sage's coercion model.
- In the single underscore methods and in ``__cmp__``, we can assume that
*both arguments belong to the same parent*. This is one benefit of the
coercion model. Note that ``__cmp__`` should be provided, since otherwise
comparison does not work in the way expected in Python::
sage: class Foo(sage.structure.element.Element):
....: def __init__(self, parent, x):
....: self.x = x
....: def _repr_(self):
....: return "<%s>"%self.x
sage: a = Foo(ZZ, 1)
sage: b = Foo(ZZ, 2)
sage: cmp(a,b)
Traceback (most recent call last):
...
NotImplementedError: BUG: sort algorithm for elements of 'None' not implemented
- When constructing new elements as the result of arithmetic operations, we do
not directly name our class, but we use ``self.__class__``. Later, this will
come in handy.
This gives rise to the following code::
sage: class MyElement(FieldElement):
....: def __init__(self, parent,n,d=None):
....: B = parent.base()
....: if d is None:
....: d = B.one_element()
....: if n not in B or d not in B:
....: raise ValueError("Numerator and denominator must be elements of %s"%B)
....: # Numerator and denominator should not just be "in" B,
....: # but should be defined as elements of B
....: d = B(d)
....: n = B(n)
....: if d==0:
....: raise ZeroDivisionError("The denominator must not be zero")
....: if d<0:
....: self.n = -n
....: self.d = -d
....: else:
....: self.n = n
....: self.d = d
....: FieldElement.__init__(self,parent)
....: def numerator(self):
....: return self.n
....: def denominator(self):
....: return self.d
....: def _repr_(self):
....: return "(%s):(%s)"%(self.n,self.d)
....: def __cmp__(self, other):
....: return cmp(self.n*other.denominator(), other.numerator()*self.d)
....: def _add_(self, other):
....: C = self.__class__
....: D = self.d*other.denominator()
....: return C(self.parent(), self.n*other.denominator()+self.d*other.numerator(), D)
....: def _sub_(self, other):
....: C = self.__class__
....: D = self.d*other.denominator()
....: return C(self.parent(), self.n*other.denominator()-self.d*other.numerator(),D)
....: def _mul_(self, other):
....: C = self.__class__
....: return C(self.parent(), self.n*other.numerator(), self.d*other.denominator())
....: def _div_(self, other):
....: C = self.__class__
....: return C(self.parent(), self.n*other.denominator(), self.d*other.numerator())
.. end of output
Features and limitations of the basic implementation
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Thanks to the single underscore methods, some basic arithmetics works, **if**
we stay inside a single parent structure::
sage: P = MyFrac(ZZ)
sage: a = MyElement(P, 3, 4)
sage: b = MyElement(P, 1, 2)
sage: a+b, a-b, a*b, a/b
((10):(8), (2):(8), (3):(8), (6):(4))
sage: a-b == MyElement(P, 1, 4)
True
.. end of output
We didn't implement exponentiation\---but it just works::
sage: a^3
(27):(64)
.. end of output
There is a default implementation of element tests. We can already do
::
sage: a in P
True
.. end of output
since `a` is defined as an element of `P`. However, we can not verify yet that
the integers are contained in the fraction field of the ring of integers. It
does not even give a wrong answer, but results in an error::
sage: 1 in P
Traceback (most recent call last):
...
NotImplementedError
.. end of output
We will take care of this later.
Categories in Sage
==================
Sometimes the base classes do not reflect the mathematics: The set of `m\times
n` matrices over a field forms, in general, not more than a vector
space. Hence, this set (called :class:`~sage.matrix.matrix_space.MatrixSpace`)
is not implemented on top of :class:`sage.rings.ring.Ring`. However, if
`m=n`, then the matrix space is an algebra, thus, is a ring.
From the point of view of Python base classes, both cases are the same::
sage: MS1 = MatrixSpace(QQ,2,3)
sage: isinstance(MS1, Ring)
False
sage: MS2 = MatrixSpace(QQ,2)
sage: isinstance(MS2, Ring)
False
.. end of output
Sage's category framework can differentiate the two cases::
sage: Rings()
Category of rings
sage: MS1 in Rings()
False
sage: MS2 in Rings()
True
.. end of output
And indeed, ``MS2`` has *more* methods than ``MS1``::
sage: import inspect
sage: len([s for s in dir(MS1) if inspect.ismethod(getattr(MS1,s,None))])
57
sage: len([s for s in dir(MS2) if inspect.ismethod(getattr(MS2,s,None))])
82
This is because the class of ``MS2`` also inherits from the parent
class for algebras::
sage: MS1.__class__.__bases__
(,
)
sage: MS2.__class__.__bases__
(,
)
.. end of output
Below, we will explain how this can be taken advantage of.
It is no surprise that our parent `P` defined above knows that it belongs to
the category of fields, as it is derived from the base class of fields.
::
sage: P.category()
Category of fields
.. end of output
However, we could choose a smaller category, namely the category of quotient fields.
Why should one choose a category?
---------------------------------
One can provide **default methods** *for all objects* of a category, and
*for all elements* of such objects. Hence, the category framework is a way
to inherit useful stuff that is not present in the base classes. These
default methods do not rely on implementation details, but on mathematical
concepts.
In addition, the categories define **test suites** for their objects and
elements\---see the last section. Hence, one also gets basic sanity tests for
free.
How does the *category framework* work?
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Abstract base classes for the objects ("parent_class") and the elements of
objects ("element_class") are provided by attributes of the category. During
initialisation of a parent, the class of the parent is *dynamically changed*
into a sub\--class of the category's parent class. Likewise, sub\--classes of
the category's element class are available for the creation of elements of the
parent, as explained below.
A dynamic change of classes does not work in Cython. Nevertheless, method
inheritance still works, by virtue of a ``__getattr__`` method.
.. NOTE::
It is strongly recommended to use the category framework both in Python
and in Cython.
Let us see whether there is any gain in chosing the category of quotient
fields instead of the category of fields::
sage: QuotientFields().parent_class, QuotientFields().element_class
(,
)
sage: [p for p in dir(QuotientFields().parent_class) if p not in dir(Fields().parent_class)]
[]
sage: [p for p in dir(QuotientFields().element_class) if p not in dir(Fields().element_class)]
['_derivative', 'denominator', 'derivative', 'factor',
'numerator', 'partial_fraction_decomposition']
.. end of output
So, there is no immediate gain for our fraction fields, but additional methods
become available to our fraction field elements. Note that some of these
methods are place-holders: There is no default implementation, but it is
*required* (respectively is *optional*) to implement these methods::
sage: QuotientFields().element_class.denominator
sage: from sage.misc.abstract_method import abstract_methods_of_class
sage: abstract_methods_of_class(QuotientFields().element_class)['optional']
['_add_', '_mul_']
sage: abstract_methods_of_class(QuotientFields().element_class)['required']
['__nonzero__', 'denominator', 'numerator']
Hence, when implementing elements of a quotient field, it is *required* to
implement methods returning the denominator and the numerator, and a method
that tells whether the element is nonzero, and in addition, it is *optional*
(but certainly recommended) to provide some arithmetic methods. If one forgets
to implement the required methods, the test suites of the category framework
will complain\---see below.
Implementing the category framework for the parent
--------------------------------------------------
We simply need to declare the correct category by an optional argument of the
field constructor, where we provide the possibility to override the default
category::
sage: from sage.categories.quotient_fields import QuotientFields
sage: class MyFrac(MyFrac):
....: def __init__(self, base, category=None):
....: if base not in IntegralDomains():
....: raise ValueError, "%s is no integral domain"%base
....: Field.__init__(self, base, category=category or QuotientFields())
When constructing instances of ``MyFrac``, their class is dynamically changed
into a new class called ``MyFrac_with_category``. It is a common sub\--class of
``MyFrac`` and of the category's parent class::
sage: P = MyFrac(ZZ)
sage: type(P)
sage: isinstance(P,MyFrac)
True
sage: isinstance(P,QuotientFields().parent_class)
True
The fraction field `P` inherits additional methods. For example, the base
class :class:`~sage.rings.fields.Field` does not have a method ``sum``. But
`P` inherits such method from the category of commutative additive
monoids\---see
:meth:`~sage.categories.commutative_additive_monoids.CommutativeAdditiveMonoids.ParentMethods.sum`::
sage: P.sum.__module__
'sage.categories.additive_monoids'
.. end of output
We have seen above that we can add elements. Nevertheless, the ``sum`` method
does not work, yet::
sage: a = MyElement(P, 3, 4)
sage: b = MyElement(P, 1, 2)
sage: c = MyElement(P, -1, 2)
sage: P.sum([a, b, c])
Traceback (most recent call last):
...
NotImplementedError
.. end of output
The reason is that the ``sum`` method starts with the return value of
``P.zero_element()``, which defaults to ``P(0)``\---but the conversion of
integers into ``P`` is not implemented, yet.
Implementing the category framework for the elements
----------------------------------------------------
Similar to what we have seen for parents, a new class is dynamically created
that combines the element class of the parent's category with the class that
we have implemented above. However, the category framework is implemented in a
different way for elements than for parents:
- We provide the parent `P` (or its class) with an attribute called
"``Element``", whose value is a class.
- The parent *automatically* obtains an attribute ``P.element_class``, that
subclasses both ``P.Element`` and ``P.category().element_class``.
Hence, for providing our fraction fields with their own element classes, **we
just need to add a single line to our class**::
sage: class MyFrac(MyFrac):
....: Element = MyElement
.. end of output
This little change provides several benefits:
- We can now create elements by simply calling the parent::
sage: P = MyFrac(ZZ)
sage: P(1), P(2,3)
((1):(1), (2):(3))
- There is a method ``zero_element`` returning the expected result::
sage: P.zero_element()
(0):(1)
- The ``sum`` method mentioned above suddenly works::
sage: a = MyElement(P, 9, 4)
sage: b = MyElement(P, 1, 2)
sage: c = MyElement(P, -1, 2)
sage: P.sum([a,b,c])
(36):(16)
.. end of output
What did happen behind the scenes to make this work?
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
We provided ``P.Element``, and thus obtain ``P.element_class``, which is a
*lazy attribute*. It provides a *dynamic* class, which is a sub\--class of
both ``MyElement`` defined above and of ``P.category().element_class``::
sage: P.__class__.element_class
sage: P.element_class
sage: type(P.element_class)
sage: issubclass(P.element_class, MyElement)
True
sage: issubclass(P.element_class,P.category().element_class)
True
.. end of output
The *default* ``__call__`` method of `P` passes the given arguments to
``P.element_class``, adding the argument ``parent=P``. This is why we are now
able to create elements by calling the parent.
In particular, these elements are instances of that new dynamic class::
sage: type(P(2,3))
.. end of output
.. NOTE::
*All* elements of `P` should use the element class. In order to make sure
that this also holds for the result of arithmetic operations, we created
them as instances of ``self.__class__`` in the arithmetic methods of
``MyElement``.
``P.zero_element()`` defaults to returning ``P(0)`` and thus returns an
instance of ``P.element_class``. Since ``P.sum([...])`` starts the summation with
``P.zero_element()`` and the class of the sum only depends on the first
summand, by our implementation, we have::
sage: type(a)
sage: isinstance(a,P.element_class)
False
sage: type(P.sum([a,b,c]))
.. end of output
The method ``factor`` provided by ``P.category().element_class`` (see above)
simply works::
sage: a; a.factor(); P(6,4).factor()
(9):(4)
2^-2 * 3^2
2^-1 * 3
.. end of output
But that's surprising: The element `a` is just an instance of ``MyElement``,
but not of ``P.element_class``, and its class does not know about the factor
method. In fact, this is due to a ``__getattr__`` method defined for
:class:`sage.structure.element.Element`.
::
sage: hasattr(type(a), 'factor')
False
sage: hasattr(P.element_class, 'factor')
True
sage: hasattr(a, 'factor')
True
.. end of output
A first note on performance
---------------------------
The category framework is sometimes blamed for speed regressions, as in
:trac:`9138` and :trac:`11900`. But if the category framework is *used
properly*, then it is fast. For illustration, we determine the time needed to
access an attribute inherited from the element class. First, we consider an
element that uses the class that we implemented above, but does not use the
category framework properly::
sage: type(a)
sage: timeit('a.factor',number=1000) # random
1000 loops, best of 3: 2 us per loop
.. end of output
Now, we consider an element that is equal to `a`, but uses the category
framework properly::
sage: a2 = P(9,4)
sage: a2 == a
True
sage: type(a2)
sage: timeit('a2.factor',number=1000) # random
1000 loops, best of 3: 365 ns per loop
.. end of output
So, *don't be afraid of using categories!*
Coercion\---the basics
======================
Theoretical background
----------------------
Coercion is not just *type conversion*
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
"Coercion" in the C programming language means "automatic type
conversion". However, in Sage, coercion is involved if one wants to be able to
do arithmetic, comparisons, etc. between elements of distinct parents. Hence,
**coercion is not about a change of types, but about a change of parents.**
As an illustration, we show that elements of the same type may very well belong
to rather different parents::
sage: P1 = QQ['v,w']; P2 = ZZ['w,v']
sage: type(P1.gen()) == type(P2.gen())
True
sage: P1 == P2
False
.. end of output
`P_2` naturally is a sub\--ring of `P_1`. So, it makes sense to be able to add
elements of the two rings\---the result should then live in `P_1`, and indeed
it does::
sage: (P1.gen()+P2.gen()).parent() is P1
True
.. end of output
It would be rather inconvenient if one needed to *manually* convert an element
of `P_2` into `P_1` before adding. The coercion system does that conversion
automatically.
Not every conversion is a coercion
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
A coercion happens implicitly, without being explicitly requested by the
user. Hence, coercion must be based on mathematical rigour. In our example,
any element of `P_2` can be naturally interpreted as an element of `P_1`. We
thus have::
sage: P1.has_coerce_map_from(P2)
True
sage: P1.coerce_map_from(P2)
Conversion map:
From: Multivariate Polynomial Ring in w, v over Integer Ring
To: Multivariate Polynomial Ring in v, w over Rational Field
While there is a conversion from `P_1` to `P_2` (namely restricted to
polynomials with integral coefficients), this conversion is not a coercion::
sage: P2.convert_map_from(P1)
Conversion map:
From: Multivariate Polynomial Ring in v, w over Rational Field
To: Multivariate Polynomial Ring in w, v over Integer Ring
sage: P2.has_coerce_map_from(P1)
False
sage: P2.coerce_map_from(P1) is None
True
.. end of output
The four axioms requested for coercions
.......................................
1. A coercion is a morphism in an appropriate category.
This first axiom has two implications:
A. A coercion is defined on all elements of a parent.
A polynomial of degree zero over the integers can be interpreted as an
integer\---but the attempt to convert a polynomial of non-zero degree
would result in an error::
sage: ZZ(P2.one())
1
sage: ZZ(P2.gen(1))
Traceback (most recent call last):
...
TypeError: not a constant polynomial
Hence, we only have a *partial* map. This is fine for a *conversion*,
but a partial map does not qualify as a *coercion*.
B. Coercions are structure preserving.
Any real number can be converted to an integer, namely by
rounding. However, such a conversion is not useful in arithmetic
operations, since the underlying algebraic structure is not preserved::
sage: int(1.6)+int(2.7) == int(1.6+2.7)
False
.. end of output
The structure that is to be preserved depends on the category of the
involved parents. For example, the coercion from the integers into the
rational field is a homomorphism of euclidean domains::
sage: QQ.coerce_map_from(ZZ).category_for()
Category of euclidean domains
.. end of output
2. There is at most one coercion from one parent to another
In addition, if there is a *coercion* from `P_2` to `P_1`, then a
*conversion* from `P_2` to `P_1` is defined for all elements of `P_2` and
coincides with the coercion.
Nonetheless, user-exposed maps are copies of the internally used maps whence
the lack of identity between different instantiations::
sage: P1.coerce_map_from(P2) is P1.convert_map_from(P2)
False
For internally used maps, the maps are identical::
sage: P1._internal_coerce_map_from(P2) is P1._internal_convert_map_from(P2)
True
.. end of output
3. Coercions can be composed
If there is a coercion `\varphi: P_1 \to P_2` and another coercion `\psi:
P_2 \to P_3`, then the composition of `\varphi` followed by `\psi` must
yield the unique coercion from `P_1` to `P_3`.
4. The identity is a coercion
Together with the two preceding axioms, it follows: If there are coercions
from `P_1` to `P_2` and from `P_2` to `P_1`, then they are mutually
inverse.
Implementing a conversion
-------------------------
We have seen above that some conversions into our fraction fields became
available after providing the attribute ``Element``. However, we can not
convert elements of a fraction field into elements of another fraction field,
yet::
sage: P(2/3)
Traceback (most recent call last):
...
ValueError: Numerator and denominator must be elements of Integer Ring
.. end of output
For implementing a conversion, **the default __call__ method should (almost)
never be overridden.** Instead, **we implement the method
_element_constructor_**, that should return an instance of the parent's
element class. Some old parent classes violate that rule\---please help to
refactor them!
::
sage: class MyFrac(MyFrac):
....: def _element_constructor_(self, *args, **kwds):
....: if len(args)!=1:
....: return self.element_class(self, *args, **kwds)
....: x = args[0]
....: try:
....: P = x.parent()
....: except AttributeError:
....: return self.element_class(self, x, **kwds)
....: if P in QuotientFields() and P != self.base():
....: return self.element_class(self, x.numerator(), x.denominator(), **kwds)
....: return self.element_class(self, x, **kwds)
.. end of output
In addition to the conversion from the base ring and from pairs of base ring
elements, we now also have a conversion from the rationals to our fraction
field of `\ZZ`:
::
sage: P = MyFrac(ZZ)
sage: P(2); P(2,3); P(3/4)
(2):(1)
(2):(3)
(3):(4)
.. end of output
Recall that above, the test `1 \in P` failed with an error. We try again and
find that the error has disappeared. This is because we are now able to
convert the integer `1` into `P`. But the containment test still yields a
wrong answer::
sage: 1 in P
False
.. end of output
The technical reason: We have a conversion `P(1)` of `1` into `P`, but this is
not known as a coercion\---yet!
::
sage: P.has_coerce_map_from(ZZ), P.has_coerce_map_from(QQ)
(False, False)
.. end of output
Establishing a coercion
-----------------------
There are two main ways to make Sage use a particular conversion as a
coercion:
- One can use :meth:`sage.structure.parent.Parent.register_coercion`, normally
during initialisation of the parent (see documentation of the method).
- A more flexible way is to provide a method ``_coerce_map_from_`` for the
parent.
Let `P` and `R` be parents. If ``P._coerce_map_from_(R)`` returns ``False``
or ``None``, then there is no coercion from `R` to `P`. If it returns a map
with domain `R` and codomain `P`, then this map is used for coercion. If it
returns ``True``, then the conversion from `R` to `P` is used as coercion.
Note that in the following implementation, we need a special case for the
rational field, since ``QQ.base()`` is not the ring of integers.
::
sage: class MyFrac(MyFrac):
....: def _coerce_map_from_(self, S):
....: if self.base().has_coerce_map_from(S):
....: return True
....: if S in QuotientFields():
....: if self.base().has_coerce_map_from(S.base()):
....: return True
....: if hasattr(S,'ring_of_integers') and self.base().has_coerce_map_from(S.ring_of_integers()):
....: return True
.. end of output
By the method above, a parent coercing into the base ring will also coerce
into the fraction field, and a fraction field coerces into another fraction
field if there is a coercion of the corresponding base rings. Now, we have::
sage: P = MyFrac(QQ['x'])
sage: P.has_coerce_map_from(ZZ['x']), P.has_coerce_map_from(Frac(ZZ['x'])), P.has_coerce_map_from(QQ)
(True, True, True)
.. end of output
We can now use coercion from `\ZZ[x]` and from `\QQ` into `P` for arithmetic
operations between the two rings::
sage: 3/4+P(2)+ZZ['x'].gen(), (P(2)+ZZ['x'].gen()).parent() is P
((4*x + 11):(4), True)
.. end of output
Equality and element containment
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Recall that above, the test `1 \in P` gave a wrong answer. Let us repeat the
test now::
sage: 1 in P
True
.. end of output
Why is that?
The default element containment test `x \in P` is based on the interplay of
three building blocks: conversion, coercion, and equality test.
#. Clearly, if the conversion `P(x)` raises an error, then `x` can not be seen as an element of `P`. On the other hand, a conversion `P(x)` can generally do very nasty things. So, the fact that `P(x)` works without error is necessary, but not sufficient for `x \in P`.
#. If `P` is the parent of `x`, then the conversion `P(x)` will not change `x` (at least, that's the default). Hence, we will have `x=P(x)`.
#. Sage uses coercion not only for arithmetic operations, but also for comparison: *If* there is a coercion from the parent of `x` to `P`, then the equality test ``x==P(x)`` reduces to ``P(x)==P(x)``. Otherwise, ``x==P(x)`` will evaluate as false.
That leads to the following default implementation of element containment testing:
.. NOTE::
`x \in P` holds if and only if the test ``x==P(x)`` does not
raise an error and evaluates as true.
If the user is not happy with that behaviour, the "magical" Python method
``__contains__`` can be overridden.
Coercion\---the advanced parts
==============================
So far, we are able to add integers and rational numbers to elements of our
new implementation of the fraction field of `\ZZ`.
::
sage: P = MyFrac(ZZ)
.. end of output
::
sage: 1/2+P(2,3)+1
(13):(6)
.. end of output
Surprisingly, we can even add a polynomial over the integers to an element of
`P`, even though the *result lives in a new parent*, namely in a polynomial
ring over `P`::
sage: P(1/2) + ZZ['x'].gen(), (P(1/2) + ZZ['x'].gen()).parent() is P['x']
((1):(1)*x + (1):(2), True)
.. end of output
In the next, seemingly more easy example, there "obviously" is a coercion from
the fraction field of `\ZZ` to the fraction field of `\ZZ[x]`. However, Sage
does not know enough about our new implementation of fraction fields. Hence,
it does not recognise the coercion::
sage: Frac(ZZ['x']).has_coerce_map_from(P)
False
.. end of output
Two obvious questions arise:
#. How / why has the new ring been constructed in the example above?
#. How can we establish a coercion from `P` to `\mathrm{Frac}(\ZZ[x])`?
The key to answering both question is the construction of parents from simpler
pieces, that we are studying now. Note that we will answer the second question
*not* by providing a coercion from `P` to `\mathrm{Frac}(\ZZ[x])`, but by
teaching Sage to automatically construct `\mathrm{MyFrac}(\ZZ[x])` and coerce
both `P` and `\mathrm{Frac}(\ZZ[x])` into it.
If we are lucky, a parent can tell how it has been constructed::
sage: Poly,R = QQ['x'].construction()
sage: Poly,R
(Poly[x], Rational Field)
sage: Fract,R = QQ.construction()
sage: Fract,R
(FractionField, Integer Ring)
In both cases, the first value returned by
:meth:`~sage.structure.parent.Parent.construction` is a mathematical
construction, called *construction functor*\---see
:class:`~sage.categories.pushout.ConstructionFunctor`. The second return value
is a simpler parent to which the construction functor is applied.
Being functors, the same construction can be applied to different objects of a
category::
sage: Poly(QQ) is QQ['x']
True
sage: Poly(ZZ) is ZZ['x']
True
sage: Poly(P) is P['x']
True
sage: Fract(QQ['x'])
Fraction Field of Univariate Polynomial Ring in x over Rational Field
Let us see on which categories these construction functors are defined::
sage: Poly.domain()
Category of rings
sage: Poly.codomain()
Category of rings
sage: Fract.domain()
Category of integral domains
sage: Fract.codomain()
Category of fields
In particular, the construction functors can be composed::
sage: Poly*Fract
Poly[x](FractionField(...))
sage: (Poly*Fract)(ZZ) is QQ[x]
True
.. end of output
In addition, it is often assumed that we have a coercion from input to output of the
construction functor::
sage: ((Poly*Fract)(ZZ)).coerce_map_from(ZZ)
Composite map:
From: Integer Ring
To: Univariate Polynomial Ring in x over Rational Field
Defn: Natural morphism:
From: Integer Ring
To: Rational Field
then
Polynomial base injection morphism:
From: Rational Field
To: Univariate Polynomial Ring in x over Rational Field
.. end of output
Construction functors do not necessarily commute::
sage: (Fract*Poly)(ZZ)
Fraction Field of Univariate Polynomial Ring in x over Integer Ring
.. end of output
The pushout of construction functors
------------------------------------
We can now formulate our problem. We have parents `P_1`, `P_2` and `R`, and
construction functors `F_1`, `F_2`, such that `P_1 = F_1(R)` and `P_2 =
F_2(R)`. We want to find a new construction functor `F_3`, such that both
`P_1` and `P_2` coerce into `P_3 = F_3(R)`.
In analogy to a notion of category theory, `P_3` is called the
:func:`~sage.categories.pushout.pushout` of `P_1` and `P_2`; and similarly
`F_3` is called the pushout of `F_1` and `F_2`.
::
sage: from sage.categories.pushout import pushout
sage: pushout(Fract(ZZ),Poly(ZZ))
Univariate Polynomial Ring in x over Rational Field
.. end of output
`F_1\circ F_2` and `F_2\circ F_1` are natural candidates for the pushout of
`F_1` and `F_2`. However, the order of the functors must rely on a canonical
choice. "Indecomposable" construction functors have a *rank*, and this allows
to order them canonically:
.. NOTE::
If ``F1.rank`` is smaller than ``F2.rank``, then the pushout is `F_2\circ
F_1` (hence, `F_1` is applied first).
We have
::
sage: Fract.rank, Poly.rank
(5, 9)
.. end of output
and thus the pushout is
::
sage: Fract.pushout(Poly), Poly.pushout(Fract)
(Poly[x](FractionField(...)), Poly[x](FractionField(...)))
.. end of output
This is why the example above has worked.
However, only "elementary" construction functors have a rank::
sage: (Fract*Poly).rank
Traceback (most recent call last):
...
AttributeError: 'CompositeConstructionFunctor' object has no attribute 'rank'
.. end of output
Shuffling composite construction functors
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
If composed construction fuctors `...\circ F_2\circ F_1` and `...\circ
G_2\circ G_1` are given, then Sage determines their pushout by *shuffling* the
constituents:
- If ``F1.rank < G1.rank`` then we apply `F_1` first, and continue with `...\circ F_3\circ F_2` and `...\circ G_2\circ G_1`.
- If ``F1.rank > G1.rank`` then we apply `G_1` first, and continue with `...\circ F_2\circ F_1` and `...\circ G_3\circ G_2`.
If ``F1.rank == G1.rank``, then the tie needs to be broken by other techniques
(see below).
As an illustration, we first get us some functors and then see how chains of
functors are shuffled.
::
sage: AlgClos, R = CC.construction(); AlgClos
AlgebraicClosureFunctor
.. end of output
::
sage: Compl, R = RR.construction(); Compl
Completion[+Infinity]
.. end of output
::
sage: Matr, R = (MatrixSpace(ZZ,3)).construction(); Matr
MatrixFunctor
.. end of output
::
sage: AlgClos.rank, Compl.rank, Fract.rank, Poly.rank, Matr.rank
(3, 4, 5, 9, 10)
.. end of output
When we apply ``Fract``, ``AlgClos``, ``Poly`` and ``Fract`` to the ring of
integers, we obtain::
sage: (Fract*Poly*AlgClos*Fract)(ZZ)
Fraction Field of Univariate Polynomial Ring in x over Algebraic Field
.. end of output
When we apply ``Compl``, ``Matr`` and ``Poly`` to the ring of integers, we
obtain::
sage: (Poly*Matr*Compl)(ZZ)
Univariate Polynomial Ring in x over Full MatrixSpace of 3 by 3 dense matrices over Real Field with 53 bits of precision
.. end of output
Applying the shuffling procedure yields
::
sage: (Poly*Matr*Fract*Poly*AlgClos*Fract*Compl)(ZZ)
Univariate Polynomial Ring in x over Full MatrixSpace of 3 by 3 dense matrices over Fraction Field of Univariate Polynomial Ring in x over Complex Field with 53 bits of precision
.. end of output
and this is indeed equal to the pushout found by Sage::
sage: pushout((Fract*Poly*AlgClos*Fract)(ZZ), (Poly*Matr*Compl)(ZZ))
Univariate Polynomial Ring in x over Full MatrixSpace of 3 by 3 dense matrices over Fraction Field of Univariate Polynomial Ring in x over Complex Field with 53 bits of precision
.. end of output
Breaking the tie
^^^^^^^^^^^^^^^^
If ``F1.rank==G1.rank`` then Sage's pushout constructions offers two ways to
proceed:
#. Construction functors have a method :meth:`~sage.categories.pushout.ConstructionFunctor.merge` that either returns ``None`` or returns a construction functor\---see below. If either ``F1.merge(G1)`` or ``G1.merge(F1)`` returns a construction functor `H_1`, then we apply `H_1` and continue with `...\circ F_3\circ F_2` and `...\circ G_3\circ G_2`.
#. Construction functors have a method :meth:`~sage.categories.pushout.ConstructionFunctor.commutes`. If either ``F1.commutes(G1)`` or ``G1.commutes(F1)`` returns ``True``, then we apply both `F_1` and `G_1` in any order, and continue with `...\circ F_3\circ F_2` and `...\circ G_3\circ G_2`.
By default, ``F1.merge(G1)`` returns ``F1`` if ``F1==G1``, and returns
``None`` otherwise. The ``commutes()`` method exists, but it seems that so far
nobody has implemented two functors of the same rank that commute.
Establishing a default implementation
-------------------------------------
The typical application of
:meth:`~sage.categories.pushout.ConstructionFunctor.merge` is to provide a
coercion between *different implementations* of the *same algebraic
structure*.
.. NOTE::
If ``F1(P)`` and ``F2(P)`` are different implementations of the same thing, then ``F1.merge(F2)(P)`` should return the default implementation.
We want to boldly turn our toy implementation of fraction fields into the new
default implementation. Hence:
- Next, we implement a new version of the "usual" fraction field functor, having the same rank, but returning our new implementation.
- We make our new implementation the default, by virtue of a merge method.
.. WARNING::
- Do not override the default ``__call__`` method of :class:`~sage.categories.pushout.ConstructionFunctor`\---implement ``_apply_functor`` instead.
- Declare domain and codomain of the functor during initialisation.
::
sage: from sage.categories.pushout import ConstructionFunctor
sage: class MyFracFunctor(ConstructionFunctor):
....: rank = 5
....: def __init__(self):
....: ConstructionFunctor.__init__(self, IntegralDomains(), Fields())
....: def _apply_functor(self, R):
....: return MyFrac(R)
....: def merge(self, other):
....: if isinstance(other, (type(self), sage.categories.pushout.FractionField)):
....: return self
.. end of output
::
sage: MyFracFunctor()
MyFracFunctor
.. end of output
We verify that our functor can really be used to construct our implementation of fraction fields, and that it can be merged with either itself or the usual fraction field constructor:
::
sage: MyFracFunctor()(ZZ)
NewFrac(Integer Ring)
.. end of output
::
sage: MyFracFunctor().merge(MyFracFunctor())
MyFracFunctor
.. end of output
::
sage: MyFracFunctor().merge(Fract)
MyFracFunctor
.. end of output
There remains to let our new fraction fields know about the new construction functor:
::
sage: class MyFrac(MyFrac):
....: def construction(self):
....: return MyFracFunctor(), self.base()
.. end of output
::
sage: MyFrac(ZZ['x']).construction()
(MyFracFunctor, Univariate Polynomial Ring in x over Integer Ring)
.. end of output
Due to merging, we have:
::
sage: pushout(MyFrac(ZZ['x']), Frac(QQ['x']))
NewFrac(Univariate Polynomial Ring in x over Rational Field)
.. end of output
A second note on performance
----------------------------
Being able to do arithmetics involving elements of different parents, with the
automatic creation of a pushout to contain the result, is certainly
convenient\---but one should not rely on it, if speed matters. Simply the
conversion of elements into different parents takes time. Moreover, by
:trac:`14058`, the pushout may be subject to Python's cyclic garbage
collection. Hence, if one does not keep a strong reference to it, the same
parent may be created repeatedly, which is a waste of time. In the following
example, we illustrate the slow\--down resulting from blindly relying on
coercion::
sage: ZZxy = ZZ['x','y']
sage: a = ZZxy('x')
sage: b = 1/2
sage: timeit("c = a+b") # random
10000 loops, best of 3: 172 us per loop
sage: QQxy = QQ['x','y']
sage: timeit("c2 = QQxy(a)+QQxy(b)") # random
10000 loops, best of 3: 168 us per loop
sage: a2 = QQxy(a)
sage: b2 = QQxy(b)
sage: timeit("c2 = a2+b2") # random
100000 loops, best of 3: 10.5 us per loop
Hence, if one avoids the explicit or implicit conversion into the pushout, but
works in the pushout right away, one can get a more than 10\--fold speed\--up.
The test suites of the category framework
=========================================
The category framework does not only provide functionality but also a test
framework. This section logically belongs to the section on categories, but
without the bits that we have implemented in the section on coercion, our
implementation of fraction fields would not have passed the tests yet.
"Abstract" methods
------------------
We have already seen above that a category can require/suggest certain parent
or element methods, that the user must/should implement. This is in order to
smoothly blend with the methods that already exist in Sage.
The methods that ought to be provided are called
:func:`~sage.misc.abstract_method.abstract_method`. Let us see what methods
are needed for quotient fields and their elements::
sage: from sage.misc.abstract_method import abstract_methods_of_class
.. end of output
::
sage: abstract_methods_of_class(QuotientFields().parent_class)['optional']
[]
sage: abstract_methods_of_class(QuotientFields().parent_class)['required']
['__contains__']
.. end of output
Hence, the only required method (that is actually required for all parents
that belong to the category of sets) is an element containment test. That's
fine, because the base class :class:`~sage.structure.parent.Parent` provides a
default containment test.
The elements have to provide more::
sage: abstract_methods_of_class(QuotientFields().element_class)['optional']
['_add_', '_mul_']
sage: abstract_methods_of_class(QuotientFields().element_class)['required']
['__nonzero__', 'denominator', 'numerator']
.. end of output
Hence, the elements must provide ``denominator()`` and ``numerator()``
methods, and must be able to tell whether they are zero or not. The base class
:class:`~sage.structure.element.Element` provides a default ``__nonzero__()``
method. In addition, the elements may provide Sage's single underscore
arithmetic methods (actually any ring element *should* provide them).
The ``_test_...`` methods
-------------------------
If a parent or element method's name start with "_test_", it gives rise to a
test in the automatic test suite. For example, it is tested
- whether a parent `P` actually is an instance of the parent class of the category of `P`,
- whether the user has implemented the required abstract methods,
- whether some defining structural properties (e.g., commutativity) hold.
For example, if one forgets to implement required methods, one obtains the
following error::
sage: class Foo(Parent):
....: Element = sage.structure.element.Element
....: def __init__(self):
....: Parent.__init__(self, category=QuotientFields())
sage: Bar = Foo()
sage: bar = Bar.element_class(Bar)
sage: bar._test_not_implemented_methods()
Traceback (most recent call last):
...
AssertionError: Not implemented method: denominator
Here are the tests that form the test suite of quotient fields::
sage: [t for t in dir(QuotientFields().parent_class) if t.startswith('_test_')]
['_test_additive_associativity',
'_test_an_element',
'_test_associativity',
'_test_characteristic',
'_test_characteristic_fields',
'_test_distributivity',
'_test_elements',
'_test_elements_eq_reflexive',
'_test_elements_eq_symmetric',
'_test_elements_eq_transitive',
'_test_elements_neq',
'_test_euclidean_degree',
'_test_one', '_test_prod',
'_test_quo_rem',
'_test_some_elements',
'_test_zero',
'_test_zero_divisors']
.. end of output
We have implemented all abstract methods (or inherit them from base classes),
we use the category framework, and we have implemented coercions. So, we are
confident that the test suite runs without an error. In fact, it does!
.. NOTE::
The following trick with the ``__main__`` module is only needed in
doctests, not in an interactive session or when defining the classes
externally.
::
sage: __main__.MyFrac = MyFrac
sage: __main__.MyElement = MyElement
sage: P = MyFrac(ZZ['x'])
sage: TestSuite(P).run()
.. end of output
Let us see what tests are actually performed::
sage: TestSuite(P).run(verbose=True)
running ._test_additive_associativity() . . . pass
running ._test_an_element() . . . pass
running ._test_associativity() . . . pass
running ._test_category() . . . pass
running ._test_characteristic() . . . pass
running ._test_characteristic_fields() . . . pass
running ._test_distributivity() . . . pass
running ._test_elements() . . .
Running the test suite of self.an_element()
running ._test_category() . . . pass
running ._test_eq() . . . pass
running ._test_nonzero_equal() . . . pass
running ._test_not_implemented_methods() . . . pass
running ._test_pickling() . . . pass
pass
running ._test_elements_eq_reflexive() . . . pass
running ._test_elements_eq_symmetric() . . . pass
running ._test_elements_eq_transitive() . . . pass
running ._test_elements_neq() . . . pass
running ._test_eq() . . . pass
running ._test_euclidean_degree() . . . pass
running ._test_not_implemented_methods() . . . pass
running ._test_one() . . . pass
running ._test_pickling() . . . pass
running ._test_prod() . . . pass
running ._test_quo_rem() . . . pass
running ._test_some_elements() . . . pass
running ._test_zero() . . . pass
running ._test_zero_divisors() . . . pass
.. end of output
Implementing a new category with additional tests
-------------------------------------------------
As one can see, tests are also performed on elements. There are methods that
return one element or a list of some elements, relying on "typical" elements
that can be found in most algebraic structures.
::
sage: P.an_element(); P.some_elements()
(2):(1)
[(2):(1)]
.. end of output
Unfortunately, the list of elements that is returned by the default method is
of length one, and that single element could also be a bit more interesting.
The method an_element relies on a method ``_an_element_()``, so, we implement
that. We also override the some_elements method.
::
sage: class MyFrac(MyFrac):
....: def _an_element_(self):
....: a = self.base().an_element()
....: b = self.base_ring().an_element()
....: if (a+b)!=0:
....: return self(a)**2/(self(a+b)**3)
....: if b != 0:
....: return self(a)/self(b)**2
....: return self(a)**2*self(b)**3
....: def some_elements(self):
....: return [self.an_element(),self(self.base().an_element()),self(self.base_ring().an_element())]
.. end of output
::
sage: P = MyFrac(ZZ['x'])
sage: P.an_element(); P.some_elements()
(x^2):(x^3 + 3*x^2 + 3*x + 1)
[(x^2):(x^3 + 3*x^2 + 3*x + 1), (x):(1), (1):(1)]
.. end of output
Now, as we have more interesting elements, we may also add a test for the
"factor" method. Recall that the method was inherited from the category, but
it appears that it is not tested.
Normally, a test for a method defined by a category should be provided by the
same category. Hence, since ``factor`` is defined in the category of quotient
fields, a test should be added there. But we won't change source code here and
will instead create a sub\--category.
Apparently, If `e` is an element of a quotient field, the product of the
factors returned by ``e.factor()`` should be equal to `e`. For forming the
product, we use the ``prod`` method, that, no surprise, is inherited from
another category::
sage: P.prod.__module__
'sage.categories.monoids'
.. end of output
When we want to create a sub\--category, we need to provide a method
:meth:`~sage.categories.category.Category.super_categories`, that returns a
list of all immediate super categories (here: category of quotient fields).
.. WARNING::
A sub\--category `S` of a category `C` is *not* implemented as a
sub\--class of ``C.__class__``! `S` becomes a sub\--category of `C` only
if ``S.super_categories()`` returns (a sub\--category of) `C`!
The parent and element methods of a category are provided as methods of
classes that are the attributes ``ParentMethods`` and ``Element Methods`` of
the category, as follows::
sage: from sage.categories.category import Category
sage: class QuotientFieldsWithTest(Category): # do *not* inherit from QuotientFields, but ...
....: def super_categories(self):
....: return [QuotientFields()] # ... declare QuotientFields as a super category!
....: class ParentMethods:
....: pass
....: class ElementMethods:
....: def _test_factorisation(self, **options):
....: P = self.parent()
....: assert self == P.prod([P(b)**e for b,e in self.factor()])
.. end of output
We provide an instance of our quotient field implementation with that new
category. Note that categories have a default ``_repr_`` method, that guesses
a good string representation from the name of the class:
``QuotientFieldsWithTest`` becomes "quotient fields with test".
.. NOTE::
The following trick with the ``__main__`` module is only needed in
doctests, not in an interactive session or when defining the classes
externally.
::
sage: __main__.MyFrac = MyFrac
sage: __main__.MyElement = MyElement
sage: __main__.QuotientFieldsWithTest = QuotientFieldsWithTest
sage: P = MyFrac(ZZ['x'], category=QuotientFieldsWithTest())
sage: P.category()
Category of quotient fields with test
.. end of output
The new test is inherited from the category. Since ``an_element()`` is returning a
complicated element, ``_test_factorisation`` is a serious test::
sage: P.an_element()._test_factorisation
.. end of output
::
sage: P.an_element().factor()
(x + 1)^-3 * x^2
.. end of output
Last, we observe that the new test has automatically become part of the test
suite. We remark that the existing tests became more serious as well, since we
made :meth:`sage.structure.parent.Parent.an_element` return something more
interesting.
::
sage: TestSuite(P).run(verbose=True)
running ._test_additive_associativity() . . . pass
running ._test_an_element() . . . pass
running ._test_associativity() . . . pass
running ._test_category() . . . pass
running ._test_characteristic() . . . pass
running ._test_characteristic_fields() . . . pass
running ._test_distributivity() . . . pass
running ._test_elements() . . .
Running the test suite of self.an_element()
running ._test_category() . . . pass
running ._test_eq() . . . pass
running ._test_factorisation() . . . pass
running ._test_nonzero_equal() . . . pass
running ._test_not_implemented_methods() . . . pass
running ._test_pickling() . . . pass
pass
running ._test_elements_eq_reflexive() . . . pass
running ._test_elements_eq_symmetric() . . . pass
running ._test_elements_eq_transitive() . . . pass
running ._test_elements_neq() . . . pass
running ._test_eq() . . . pass
running ._test_euclidean_degree() . . . pass
running ._test_not_implemented_methods() . . . pass
running ._test_one() . . . pass
running ._test_pickling() . . . pass
running ._test_prod() . . . pass
running ._test_quo_rem() . . . pass
running ._test_some_elements() . . . pass
running ._test_zero() . . . pass
running ._test_zero_divisors() . . . pass
.. end of output
Appendix: The complete code
===========================
.. highlight:: python
:linenothreshold: 2
::
# Importing base classes, ...
import sage
from sage.rings.ring import Field
from sage.structure.element import FieldElement
from sage.categories.category import Category
# ... the UniqueRepresentation tool,
from sage.structure.unique_representation import UniqueRepresentation
# ... some categories, and ...
from sage.categories.fields import Fields
from sage.categories.quotient_fields import QuotientFields
from sage.categories.integral_domains import IntegralDomains
# construction functors
from sage.categories.pushout import ConstructionFunctor
# Fraction field elements
class MyElement(FieldElement):
def __init__(self, parent, n, d=None):
if parent is None:
raise ValueError("The parent must be provided")
B = parent.base()
if d is None:
# The default denominator is one
d = B.one_element()
# verify that both numerator and denominator belong to the base
if n not in B or d not in B:
raise ValueError("Numerator and denominator must be elements of %s"%B)
# Numerator and denominator should not just be "in" B,
# but should be defined as elements of B
d = B(d)
n = B(n)
# the denominator must not be zero
if d==0:
raise ZeroDivisionError("The denominator must not be zero")
# normalize the denominator: WLOG, it shall be non-negative.
if d<0:
self.n = -n
self.d = -d
else:
self.n = n
self.d = d
FieldElement.__init__(self,parent)
# Methods required by the category of fraction fields:
def numerator(self):
return self.n
def denominator(self):
return self.d
# String representation (single underscore!)
def _repr_(self):
return "(%s):(%s)"%(self.n,self.d)
# Comparison: We can assume that both arguments are coerced
# into the same parent, which is a fraction field. Hence, we
# are allowed to use the denominator() and numerator() methods
# on the second argument.
def __cmp__(self, other):
return cmp(self.n*other.denominator(), other.numerator()*self.d)
# Arithmetic methods, single underscore. We can assume that both
# arguments are coerced into the same parent.
# We return instances of self.__class__, because self.__class__ will
# eventually be a sub-class of MyElement.
def _add_(self, other):
C = self.__class__
D = self.d*other.denominator()
return C(self.parent(), self.n*other.denominator()+self.d*other.numerator(),D)
def _sub_(self, other):
C = self.__class__
D = self.d*other.denominator()
return C(self.parent(), self.n*other.denominator()-self.d*other.numerator(),D)
def _mul_(self, other):
C = self.__class__
return C(self.parent(), self.n*other.numerator(), self.d*other.denominator())
def _div_(self, other):
C = self.__class__
return C(self.parent(), self.n*other.denominator(), self.d*other.numerator())
# Inheritance from UniqueRepresentation implements the unique parent
# behaviour. Moreover, it implements pickling (provided that Python
# succeeds to look up the class definition).
class MyFrac(UniqueRepresentation, Field):
# Implement the category framework for elements, which also
# makes some basic conversions work.
Element = MyElement
# Allow to pass to a different category, by an optional argument
def __init__(self, base, category=None):
# Fraction fields only exist for integral domains
if base not in IntegralDomains():
raise ValueError, "%s is no integral domain"%base
# Implement the category framework for the parent
Field.__init__(self, base, category=category or QuotientFields())
# Single-underscore method for string representation
def _repr_(self):
return "NewFrac(%s)"%repr(self.base())
# Two methods that are implicitly used in some tests
def base_ring(self):
return self.base().base_ring()
def characteristic(self):
return self.base().characteristic()
# Implement conversions. Do not override __call__!
def _element_constructor_(self, *args, **kwds):
if len(args)!=1:
return self.element_class(self, *args, **kwds)
x = args[0]
try:
P = x.parent()
except AttributeError:
return self.element_class(self, x, **kwds)
if P in QuotientFields() and P != self.base():
return self.element_class(self, x.numerator(), x.denominator(), **kwds)
return self.element_class(self, x, **kwds)
# Implement coercion from the base and from fraction fields
# over a ring that coerces into the base
def _coerce_map_from_(self, S):
if self.base().has_coerce_map_from(S):
return True
if S in QuotientFields():
if self.base().has_coerce_map_from(S.base()):
return True
if hasattr(S,'ring_of_integers') and self.base().has_coerce_map_from(S.ring_of_integers()):
return True
# Tell how this parent was constructed, in order to enable pushout constructions
def construction(self):
return MyFracFunctor(), self.base()
# return some elements of this parent
def _an_element_(self):
a = self.base().an_element()
b = self.base_ring().an_element()
if (a+b)!=0:
return self(a)**2/(self(a+b)**3)
if b != 0:
return self(a)/self(b)**2
return self(a)**2*self(b)**3
def some_elements(self):
return [self.an_element(),self(self.base().an_element()),self(self.base_ring().an_element())]
# A construction functor for our implementation of fraction fields
class MyFracFunctor(ConstructionFunctor):
# The rank is the same for Sage's original fraction field functor
rank = 5
def __init__(self):
# The fraction field construction is a functor
# from the category of integral domains into the category of
# fields
# NOTE: We could actually narrow the codomain and use the
# category QuotientFields()
ConstructionFunctor.__init__(self, IntegralDomains(), Fields())
# Applying the functor to an object. Do not override __call__!
def _apply_functor(self, R):
return MyFrac(R)
# Note: To apply the functor to morphisms, implement
# _apply_functor_to_morphism
# Make sure that arithmetic involving elements of Frac(R) and
# MyFrac(R) works and yields elements of MyFrac(R)
def merge(self, other):
if isinstance(other, (type(self), sage.categories.pushout.FractionField)):
return self
# A quotient field category with additional tests.
# Notes:
# - Category inherits from UniqueRepresentation. Hence, there
# is only one category for given arguments.
# - Since QuotientFieldsWithTest is a singleton (there is only
# one instance of this class), we could inherit from
# sage.categories.category_singleton.Category_singleton
# rather than from sage.categories.category.Category
class QuotientFieldsWithTest(Category):
# Our category is a sub-category of the category of quotient fields,
# by means of the following method.
def super_categories(self):
return [QuotientFields()]
# Here, we could implement methods that are available for
# all objects in this category.
class ParentMethods:
pass
# Here, we add a new test that is available for all elements
# of any object in this category.
class ElementMethods:
def _test_factorisation(self, **options):
P = self.parent()
# The methods prod() and factor() are inherited from
# some other categories.
assert self == P.prod([P(b)**e for b,e in self.factor()])
.. highlight:: python
:linenothreshold: 22222