Exemplo n.º 1
0
    def test_json(self):
        lattice = Lattice([16, 8], 2)
        subsystem = CustomSubsystem(lambda site: site.bs[0] == 0 and site.bs[1] < 4, lattice)
        assert (
            json.dumps(subsystem.to_json())
            == '{"type": "CustomSubsystem", "_site_indices": [0, 16, 32, 48, 128, 144, 160, 176]}'
        )

        assert Subsystem.from_json(subsystem.to_json(), lattice) == subsystem
        assert CustomSubsystem.from_json(subsystem.to_json(), lattice) == subsystem
Exemplo n.º 2
0
    def test_custom_subsystem(self):
        lattice = Lattice([8, 8], 2)
        subsystem = CustomSubsystem(lambda site: site.bs[0] % 2 == 0, lattice)

        inside = set()
        for site in lattice:
            if site in subsystem:
                inside.add(site)
            else:
                with pytest.raises(ValueError):
                    subsystem.index(site)

        assert len(inside) == len(subsystem)

        for i, site in enumerate(subsystem):
            assert subsystem[i] == site
            assert subsystem.index(site) == i
            inside.remove(site)

        assert len(inside) == 0