test_data_ingest.py 464 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import unittest
import matplotlib.pyplot as plt


class TestData(unittest.TestCase):

    def test_ingest_data(self):
        from covid.pydata import ingest_data
        x = ingest_data('../data/lad_shp/Local_Authority_Districts_December_2019_Boundaries_UK_BFC.shp',
                        '../data/ukmidyearestimates20182019ladcodes.csv')
        x['geo'].plot(column='[80,inf)', legend=True)
        plt.show()


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