Example #1
0
    def testStateCliCall(self):
        sys.argv = self.st_args
        sys.stdout = io.StringIO()
        addfips_cli.main()
        sys.stdout.seek(0)
        reader = csv.DictReader(sys.stdout)
        row = next(reader)

        assert row["name"] == "Alabama"
        assert row["fips"] == "01"
Example #2
0
    def testStateNameCliCall(self):
        sys.argv = self.co_args[:-2] + ["--state-name", "Alabama"]
        sys.stdout = io.StringIO()
        addfips_cli.main()
        sys.stdout.seek(0)
        reader = csv.DictReader(sys.stdout)
        row = next(reader)

        assert row["name"] == "Autauga County"
        assert row["fips"] == "01001"
Example #3
0
    def testCountyCliCall(self):
        sys.argv = self.co_args
        sys.stdout = io.StringIO()
        addfips_cli.main()
        sys.stdout.seek(0)
        reader = csv.DictReader(sys.stdout)
        row = next(reader)

        assert row["name"] == "Autauga County"
        assert row["fips"] == "01001"
Example #4
0
    def testStateCliCallNoHeader(self):
        sys.argv = self.st_args[:2] + ["-s", "1", "--no-header"]
        sys.stdout = io.StringIO()
        addfips_cli.main()
        sys.stdout.seek(0)
        reader = csv.reader(sys.stdout)
        next(reader)
        row = next(reader)

        assert row[1] == "Alabama"
        assert row[0] == "01"