test_causetplotting.py

#!/usr/bin/env python
'''
Created on 24 Oct 2020

@author: Christoph Minz
@license: BSD 3-Clause
'''
from __future__ import annotations
from typing import List
import unittest
from causets.embeddedcauset import EmbeddedCauset  # @UnresolvedImport
from causets.sprinkledcauset import SprinkledCauset  # @UnresolvedImport
from causets.spacetimes import BlackHoleSpacetime  # @UnresolvedImport
from causets.shapes import CoordinateShape  # @UnresolvedImport
import causets.causetplotting as cplt  # @UnresolvedImport


class TestCausetplotting(unittest.TestCase):

    def setUp(self):
        cplt.setDefaultColors('UniYork')
        cplt.figure(figsize=(8.0, 8.0))

    def tearDown(self):
        pass

    def test_plotGeneric(self):
        #
        # Warning:
        # The plotting functions need to be remastered. The newer version of 
        # matplotlib does not accept (partially) empty patches anymore so that 
        # the light cones in this text case cause errors. Empty objects should 
        # be skipped and not rendered.
        #
        C: EmbeddedCauset = EmbeddedCauset(
            shape=CoordinateShape(3, 'cuboid', edges=[4.0, 2.0, 2.0]),
            coordinates=[[-1.5, 0.0, -0.3], [-0.5, 0.0, -0.1],
                         [0.5, 0.0, 0.1], [1.5, 0.0, 0.3]])
        dims: List[int] = [1, 2, 0]
        plotAxes = cplt.createAxes(True)
        P = cplt.Plotter(plotAxes, C, dims=dims,
                         pastcones={'facecolor': 'none', 'alpha': 0.8},
                         futurecones={'facecolor': 'none', 'alpha': 0.8})
        P([0.72])
        if len(dims) > 2:
            plotAxes.xaxis.pane.fill = False
            plotAxes.yaxis.pane.fill = False
            plotAxes.zaxis.pane.fill = False
        cplt.show()

    def test_plotSprinkle(self):
        C: SprinkledCauset = SprinkledCauset(card=200,
                                             spacetime=BlackHoleSpacetime(2))
        dims = [1, 0]
        e = C.PastInf.copy().pop()
        events_Cone = e.Cone
        plotAxes = cplt.createAxes()
        cplt.plot(plotAxes, C, dims=dims,
                  events={'alpha': 0.05},
                  links=False, labels=False)
        cplt.plot(plotAxes, list(events_Cone),
                  dims=dims, spacetime=C.Spacetime,
                  events={'markerfacecolor': 'cs:darkblue'},
                  links=True, labels=False)
        cplt.plot(plotAxes, e, dims=dims, spacetime=C.Spacetime,
                  events={'markerfacecolor': 'cs:red'},
                  pastcones={'alpha': 1.0,
                             'linewidth': 2.0},
                  futurecones={'alpha': 1.0,
                               'linewidth': 2.0},
                  time=[-1.0, 1.0])
        C.Shape.plot(plotAxes, dims)
        cplt.show()


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