Example #1
0
    def test_invariance(self):
        """unit-sphere locations should remain unchanged."""
        locs = [[1, 0, 0], [-1, 0, 0], [0, 1, 0], [0, -1, 0], [0, 0, 1],
                [0, 0, -1]]
        warp1 = warp_locations(locs)
        warp2, c, r = warp_locations(locs, return_ellipsoid=True)

        assert_allclose(warp1, locs, atol=1e-12)
        assert_allclose(warp2, locs, atol=1e-12)
        assert_allclose(c, 0, atol=1e-12)
        assert_allclose(r, 1, atol=1e-12)
Example #2
0
    def test_eeglocations(self):
        np.random.seed(42)

        scale = np.random.rand(3) * 10 + 10
        displace = np.random.randn(3) * 100
        noise = np.random.randn(len(eeglocs), 3) * 5

        assert_allclose(warp_locations(eeglocs), eeglocs, atol=1e-10)
        assert_allclose(warp_locations(eeglocs * scale), eeglocs, atol=1e-10)
        assert_allclose(warp_locations(eeglocs * scale + displace), eeglocs, atol=1e-10)
        warp = warp_locations(eeglocs * scale + displace + noise)
        assert_allclose(np.sum(warp**2, axis=1), 1, atol=1e-12)  # all locations on unit shpere
Example #3
0
    def test_invariance(self):
        """unit-sphere locations should remain unchanged."""
        locs = [[1, 0, 0], [-1, 0, 0],
                [0, 1, 0], [0, -1, 0],
                [0, 0, 1], [0, 0, -1]]
        warp1 = warp_locations(locs)
        warp2, c, r = warp_locations(locs, return_ellipsoid=True)

        assert_allclose(warp1, locs, atol=1e-12)
        assert_allclose(warp2, locs, atol=1e-12)
        assert_allclose(c, 0, atol=1e-12)
        assert_allclose(r, 1, atol=1e-12)
Example #4
0
    def test_eeglocations(self):
        np.random.seed(42)

        scale = np.random.rand(3) * 10 + 10
        displace = np.random.randn(3) * 100
        noise = np.random.randn(len(eeglocs), 3) * 5

        assert_allclose(warp_locations(eeglocs), eeglocs, atol=1e-10)
        assert_allclose(warp_locations(eeglocs * scale), eeglocs, atol=1e-10)
        assert_allclose(warp_locations(eeglocs * scale + displace),
                        eeglocs,
                        atol=1e-10)
        warp = warp_locations(eeglocs * scale + displace + noise)
        assert_allclose(np.sum(warp**2, axis=1), 1,
                        atol=1e-12)  # all locations on unit shpere
Example #5
0
 def test_interface(self):
     self.assertRaises(TypeError, warp_locations)
     self.assertEqual(warp_locations(np.eye(3)).shape, (3, 3))  # returns array
     self.assertEqual(len(warp_locations(np.eye(3), return_ellipsoid=True)), 3)  # returns tuple
Example #6
0
 def test_interface(self):
     self.assertRaises(TypeError, warp_locations)
     self.assertEqual(warp_locations(np.eye(3)).shape,
                      (3, 3))  # returns array
     self.assertEqual(len(warp_locations(np.eye(3), return_ellipsoid=True)),
                      3)  # returns tuple