initial commit

.gitignore

@@ -0,0 +1,2 @@
.*coverage*
__pycache__/*

Paintball.py

@@ -0,0 +1,34 @@
"""
Simple class for the paintball marker.
"""
class Marker():
    """
    A paintball marker.
    """
    def __init__(self):
        self.ammo = 0
        self.co2 = 1.0
    def fire(self):
        # Check to see if we can even fire.
        if self.ammo == 0 or self.co2 == 0.0:
            return
        self.ammo -= 1
        self.co2 -= 0.01
    def semiBurst(self):
        for round in range(0, 3):
            self.fire()
    def fullBurst(self, seconds):
        for second in range(0, seconds):
            self.fire()
    def loadHopper(self, ammo):
        self.ammo += ammo
    def refillAir(self):
        self.co2 = 1.0

PaintballTest.py

@@ -0,0 +1,35 @@
"""
Simple test of the paintball marker.
"""
import unittest
from Paintball import Marker
class TestPaintball(unittest.TestCase):
    def setUp(self):
        self.marker = Marker()
    def test_fire(self):
        self.marker.loadHopper(1)
        self.marker.fire()
        self.assertEqual(0, self.marker.ammo)
    def test_semiBurst(self):
        self.marker.loadHopper(3)
        self.marker.semiBurst()
        self.assertEqual(0, self.marker.ammo)
    def test_fullBurst(self):
        self.marker.loadHopper(5)
        self.marker.fullBurst(4)
        self.assertEqual(1, self.marker.ammo)
        self.marker.fullBurst(10)
        self.assertEqual(0, self.marker.ammo)
    def test_refillAir(self):
        self.marker.refillAir()
        self.assertEqual(1.0, self.marker.co2)
if __name__ == "__main__": #pragma: no cover
    unittest.main()

README.md

@@ -0,0 +1,16 @@
Demonstration
=============
This is a brief demonstration of the [vim-umbrella](https://github.com/revolvingcow/vim-umbrella) ViM
plugin. The files `umbrella` and `umbrella.py` were pulled directly from the plugin examples folder.
Screencast
==========
A recorded session of this demonstration may be viewed on [YouTube](http://youtu.be/qL_fD8DOfh0).
Credits
=======
Screencast: Bryan Allred ([www.enderspsyche.com](http://enderspsyche.com))
Plugin: Revolving Cow, LLC ([www.revolvingcow.com](http://www.revolvingcow.com))

umbrella

@@ -0,0 +1 @@
python umbrella.py &> /dev/null

umbrella.py

@@ -0,0 +1,40 @@
#!/usr/bin/python
import coverage
import os
import unittest
# Store our output to the file
output = ""
# Start the coverage
cov = coverage.coverage()
cov.start()
# Dynamically get all the test cases we can find.
suite = unittest.TestSuite()
suite.addTests(unittest.TestLoader().discover(os.getcwd(), pattern="*.py"))
# Run the tests
unittest.TextTestRunner().run(suite)
# Stop coverage and save the stats
cov.stop()
cov.save()
# Iterate through the tested files
for source in cov.data.measured_files():
    for a in [cov.analysis2(source)]:
        output += "{0};{1};{2};{3};\n".format(
                a[0], 
                ",".join(str(x) for x in a[1]), 
                ",".join(str(x) for x in a[2]), 
                ",".join(str(x) for x in a[3]))
# Write the coverage report for umbrella
fileName = ".umbrella-coverage"
if os.path.isfile(fileName):
    os.remove(fileName)
f = open(fileName, "w")
f.write(output)
f.close()